corrections after unit test