Minor, add return