fixes unit test