Add forgotten return type