fix source before unit test