Fixed a type problem