After being questioned by Conservative leader Kemi Badenoch, the PM says the government will look to close the "legal loophole".