Use correct contexts for Set Integration Test
Maps, Flags, and Registers now work.