Missing assignment fixed