fix to unit test