Fixed another missing assign