Fixed a return type