fix warning and unit test