unit test fixed