Fix missing assignment