fix unit problem