Return as the last statement

Be compatible with older go releases.