diff --git a/OpenVPN Adapter Tests/CustomFlow.swift b/OpenVPN Adapter Tests/CustomFlow.swift
index dbd423f..23e0f58 100644
--- a/OpenVPN Adapter Tests/CustomFlow.swift	
+++ b/OpenVPN Adapter Tests/CustomFlow.swift	
@@ -6,22 +6,15 @@
 //
 
 import NetworkExtension
+@testable import OpenVPNAdapter
 
-class CustomFlow: NEPacketTunnelFlow {
-
-    override func readPackets(completionHandler: @escaping ([Data], [NSNumber]) -> Void) {
+class CustomFlow: NSObject, OpenVPNAdapterPacketFlow {
+    
+    func readPackets(completionHandler: @escaping ([Data], [NSNumber]) -> Void) {
         
     }
     
-    override func writePackets(_ packets: [Data], withProtocols protocols: [NSNumber]) -> Bool {
-        return true
-    }
-    
-    override func readPacketObjects(completionHandler: @escaping ([NEPacket]) -> Void) {
-        
-    }
-    
-    override func writePacketObjects(_ packets: [NEPacket]) -> Bool {
+    func writePackets(_ packets: [Data], withProtocols protocols: [NSNumber]) -> Bool {
         return true
     }
     
diff --git a/OpenVPN Adapter Tests/OpenVPNAdapterTests.swift b/OpenVPN Adapter Tests/OpenVPNAdapterTests.swift
index 0642e45..40a83f6 100644
--- a/OpenVPN Adapter Tests/OpenVPNAdapterTests.swift	
+++ b/OpenVPN Adapter Tests/OpenVPNAdapterTests.swift	
@@ -135,7 +135,7 @@ class OpenVPNAdapterTests: XCTestCase {
 
 extension OpenVPNAdapterTests: OpenVPNAdapterDelegate {
     
-    func openVPNAdapter(_ openVPNAdapter: OpenVPNAdapter, configureTunnelWithNetworkSettings networkSettings: NEPacketTunnelNetworkSettings, completionHandler: @escaping (NEPacketTunnelFlow?) -> Void) {
+    func openVPNAdapter(_ openVPNAdapter: OpenVPNAdapter, configureTunnelWithNetworkSettings networkSettings: NEPacketTunnelNetworkSettings, completionHandler: @escaping (OpenVPNAdapterPacketFlow?) -> Void) {
         completionHandler(customFlow)
     }