Forgot to add the tests for the previous listcomp commit