Small fix to the previous commit

Someone should make the commented tests pass