diff --git a/driver/main.cpp b/driver/main.cpp
index 12a10ed..fd8400a 100644
--- a/driver/main.cpp
+++ b/driver/main.cpp
@@ -80,7 +80,7 @@ int main(int argc, char* argv[])
 
         nod::Mkdir(outDir, 0755);
 
-        nod::Partition* dataPart = disc->getDataPartition();
+        nod::IPartition* dataPart = disc->getDataPartition();
         if (!dataPart)
             return 1;