fixed unit test (sorry!)