fixed test (I'm bad at writing tests....)