fix test (woo! TDD!)