state("Mad2")
{
    byte loadFlag : "TfbHavokLibrary.dll", 0x342140;
}

init
{
    vars.delayFrames = 0;
    vars.wasLoading = false;
}

update
{
    bool currentlyLoading = current.loadFlag != 1;

    // Detect transition from loading to not loading
    if (vars.wasLoading && !currentlyLoading)
    {
        // Start 9-frame delay
        vars.delayFrames = 9;
    }

    vars.wasLoading = currentlyLoading;

    // Count down the delay if it's active
    if (vars.delayFrames > 0)
    {
        vars.delayFrames--;
    }
}

isLoading
{
    // Still loading OR delay is active
    return current.loadFlag != 1 || vars.delayFrames > 0;
}