- Fixed unit test