Change return type to correct value.