fixed unit test errors