I can get rid of the TODO or the if statement if you want. (All the tests pass with the check removed)