Update variable assignment to not break