Fix missing assign