Missing return fix