unit test for fixed code