Fixed missing return type