unit fixed bug