diff --git a/test/unit/Node/DirectoryTest.php b/test/unit/Node/DirectoryTest.php
index 8ca9f69..9d86648 100644
--- a/test/unit/Node/DirectoryTest.php
+++ b/test/unit/Node/DirectoryTest.php
@@ -120,7 +120,7 @@ class DirectoryTest extends UnitTestCase
     {
         $dir = new Directory();
 
-        $this->assertEquals(StatInterface::TYPE_DIR, $dir->getMode() & StatInterface::S_IFMT);
+        $this->assertEquals(StatInterface::TYPE_DIR, $dir->getMode() & StatInterface::TYPE_MASK);
     }
 
     public function testGetSize()
diff --git a/test/unit/Node/FileTest.php b/test/unit/Node/FileTest.php
index b7b0dee..2cde4ac 100644
--- a/test/unit/Node/FileTest.php
+++ b/test/unit/Node/FileTest.php
@@ -54,7 +54,7 @@ class FileTest extends UnitTestCase
     {
         $file = new File();
 
-        $this->assertEquals(StatInterface::TYPE_FILE, $file->getMode() & StatInterface::S_IFMT);
+        $this->assertEquals(StatInterface::TYPE_FILE, $file->getMode() & StatInterface::TYPE_MASK);
     }
 
     public function testGetSize()