Return fix

Forgot a return this