diff --git a/driver/main.cpp b/driver/main.cpp
index 2bb31f8..b6f44b4 100644
--- a/driver/main.cpp
+++ b/driver/main.cpp
@@ -54,7 +54,7 @@ int main(int argc, char* argv[])
         if (!disc)
             return -1;
 
-        NOD::DiscBase::IPartition* dataPart = disc->getDataPartition();
+        NOD::Partition* dataPart = disc->getDataPartition();
         if (!dataPart)
             return -1;
 
diff --git a/include/NOD/DiscBase.hpp b/include/NOD/DiscBase.hpp
index d393745..149b53c 100644
--- a/include/NOD/DiscBase.hpp
+++ b/include/NOD/DiscBase.hpp
@@ -253,6 +253,9 @@ public:
     }
 };
 
+using Partition = DiscBase::IPartition;
+using Node = Partition::Node;
+
 }
 
 #endif // __NOD_DISC_BASE__