diff --git a/test/tsan/Darwin/malloc_size.mm b/test/tsan/Darwin/malloc_size.mm
index 3d3629b..485d85b 100644
--- a/test/tsan/Darwin/malloc_size.mm
+++ b/test/tsan/Darwin/malloc_size.mm
@@ -3,8 +3,10 @@
 // RUN: %clang_tsan %s -o %t -framework Foundation
 // RUN: %run %t 2>&1 | FileCheck %s
 
-#import <Foundation/Foundation.h>
 #include <malloc/malloc.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <sys/mman.h>
 
 int some_global;
 
diff --git a/test/tsan/Darwin/xpc-race.mm b/test/tsan/Darwin/xpc-race.mm
index 957613f..9141da4 100644
--- a/test/tsan/Darwin/xpc-race.mm
+++ b/test/tsan/Darwin/xpc-race.mm
@@ -2,6 +2,7 @@
 // RUN: %env_tsan_opts=ignore_interceptors_accesses=1 %deflake %run %t 2>&1 | FileCheck %s
 
 #import <Foundation/Foundation.h>
+#import <xpc/xpc.h>
 
 #import "../test.h"
 
@@ -73,8 +74,8 @@
 // CHECK: Hello world.
 // CHECK: WARNING: ThreadSanitizer: data race
 // CHECK:   Write of size 8
-// CHECK:     #0 {{.*}}xpc-race.mm:33
+// CHECK:     #0 {{.*}}xpc-race.mm:34
 // CHECK:   Previous write of size 8
-// CHECK:     #0 {{.*}}xpc-race.mm:33
+// CHECK:     #0 {{.*}}xpc-race.mm:34
 // CHECK: Location is global 'global'
 // CHECK: Done.
diff --git a/test/tsan/Darwin/xpc.mm b/test/tsan/Darwin/xpc.mm
index c34b948..a939b02 100644
--- a/test/tsan/Darwin/xpc.mm
+++ b/test/tsan/Darwin/xpc.mm
@@ -2,6 +2,7 @@
 // RUN: %env_tsan_opts=ignore_interceptors_accesses=1 %run %t 2>&1 | FileCheck %s
 
 #import <Foundation/Foundation.h>
+#import <xpc/xpc.h>
 
 long global;