Fixed a type mistake