diff --git a/tests/functional/pyocf/types/core.py b/tests/functional/pyocf/types/core.py
index 9ed5bf6..ce90bb8 100644
--- a/tests/functional/pyocf/types/core.py
+++ b/tests/functional/pyocf/types/core.py
@@ -166,3 +166,5 @@ lib = OcfLib.getInstance()
 lib.ocf_core_get_volume.restype = c_void_p
 lib.ocf_volume_new_io.argtypes = [c_void_p]
 lib.ocf_volume_new_io.restype = c_void_p
+lib.ocf_core_get_volume.argtypes = [c_void_p]
+lib.ocf_core_get_volume.restype = c_void_p