Corrected missing return