diff --git a/Tests/DeviceFamilyTests.swift b/Tests/DeviceFamilyTests.swift
index 8a83d79..f4cc8d6 100644
--- a/Tests/DeviceFamilyTests.swift
+++ b/Tests/DeviceFamilyTests.swift
@@ -48,7 +48,7 @@ class DeviceFamilyTests: XCTestCase {
     
     func testDeviceIsSimulator() {
         let deviceFamily = DeviceFamily(rawValue: "iPhone")
-        #if arch(i386) || arch(x86_64)
+        #if targetEnvironment(simulator)
         XCTAssert(deviceFamily.isSimulator, "DeviceFamily - .isSimulator is failing")
         #else
         XCTAssert(!(deviceFamily.isSimulator), "DeviceFamily - .isSimulator is failing")