Test rig updates