fix unit issue