fixed unit test (for now)