Fix missing return statement