Fixed missing return value