Fixed error with missing return