SIOC - MIP LIGHT TEST

I am using the SIOC script from RKSoftware to interface to the PMDG 737. The SIOC script doesn't function for the LIGHT TEST. I added the following code:

IF &LIGHTS_TEST = 1
{
// Set all MIP lights
&belowGS_0 = 1
...
}
ELSE
{
...
}

The LIGHT TEST does illuminate all the LEDs. The issue I have now is that when the light test switch is turned from ON to OFF, all the LEDs now turn off. I need the LEDs that were illuminate before the LIGHT TEST to illuminate.

What is the best way to handle this? Should I save off the current setting of the LED before I illuminate the LEDs?
Example:
IF &LIGHTS_TEST = 1
{
// Save off settings of MIP and MCP LEDs
// Set all MIP lights
&belowGS_0 = 1
...
}

// Restore settings of MIP and MCP before LIGHT TEST.