unit test now fully work