diff --git a/python/pyarmnn/test/test_runtime.py b/python/pyarmnn/test/test_runtime.py
index e558e84..a37772c 100644
--- a/python/pyarmnn/test/test_runtime.py
+++ b/python/pyarmnn/test/test_runtime.py
@@ -156,7 +156,9 @@
     opt_network, _ = ann.Optimize(network, preferred_backends,
                                   runtime.GetDeviceSpec(), ann.OptimizerOptions())
 
-    properties = ann.INetworkProperties(True, True)
+    inputSource = ann.MemorySource_Malloc
+    outputSource = ann.MemorySource_Malloc
+    properties = ann.INetworkProperties(False, inputSource, outputSource)
     net_id, messages = runtime.LoadNetwork(opt_network, properties)
     assert "" == messages
     assert net_id == 0
@@ -184,11 +186,6 @@
     assert net_id == 0
 
 
-def test_network_properties_deprecated_constructor():
-    with pytest.warns(DeprecationWarning):
-        warnings.warn("Deprecated: Use constructor with MemorySource argument instead.", DeprecationWarning)
-
-
 def test_unload_network_fails_for_invalid_net_id(random_runtime):
     preferred_backends = random_runtime[0]
     network = random_runtime[1]