Fixed missing return var