fixed unitTest