minor return fix