diff --git a/tests/pylibfdt_tests.py b/tests/pylibfdt_tests.py
index 08c4f57..f195a34 100644
--- a/tests/pylibfdt_tests.py
+++ b/tests/pylibfdt_tests.py
@@ -336,10 +336,9 @@ class PyLibfdtTests(unittest.TestCase):
 
     def testIntegers(self):
         """Check that integers can be passed and returned"""
-        self.assertEquals(0, libfdt.fdt_get_phandle(self.fdt.as_bytearray(), 0))
+        self.assertEquals(0, libfdt.fdt_get_phandle(self.fdt._fdt, 0))
         node2 = self.fdt.path_offset('/subnode@2')
-        self.assertEquals(
-                0x2000, libfdt.fdt_get_phandle(self.fdt.as_bytearray(), node2))
+        self.assertEquals(0x2000, libfdt.fdt_get_phandle(self.fdt._fdt, node2))
 
     def testGetPhandle(self):
         """Test for the get_phandle() method"""