#244 fixed unit test