fixed missing $return declaration