Fixed bug in return type.