Fixed missing assignment