Set proper return