Add missing return statement. Oops