Fix the type error