Fix test

This should be the reverse lol, but I haven't really been coding my TDD.