Add forgotten return statement

Came by this "issue" by accident. Hope it helps.