From 14f2dcae1abcafd3c3b5a4ff5b9755f1c7c88c0c Mon Sep 17 00:00:00 2001
From: Peter Evans <pevans@activecampaign.com>
Date: Wed, 3 Jan 2018 15:21:24 -0600
Subject: [PATCH] Set the map machine on setup; revert to NULL on teardown

---
 tests/apple2.mem.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/apple2.mem.c b/tests/apple2.mem.c
index 62fb82f..afdc9e0 100644
--- a/tests/apple2.mem.c
+++ b/tests/apple2.mem.c
@@ -10,12 +10,14 @@ setup()
 {
     mach = apple2_create(100, 100);
     apple2_mem_map(mach);
+    vm_segment_set_map_machine(mach);
 }
 
 static void
 teardown()
 {
     apple2_free(mach);
+    vm_segment_set_map_machine(NULL);
 }
 
 TestSuite(apple2_mem, .init = setup, .fini = teardown);