Fixed stupid return error