Fixed the add type error