We've deleted the 'TODO we need to add a test' but not added the test?