Added missing assign