changed to pass unit test