diff --git a/src/Lib.cpp b/src/Lib.cpp
index 1f2cc79..8ca429b 100644
--- a/src/Lib.cpp
+++ b/src/Lib.cpp
@@ -18,6 +18,7 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <libpmt/lib.hpp>
+#include "functions/functions.hpp"
 
 namespace PartitionManager {
 
diff --git a/src/functions/functions.hpp b/src/functions/functions.hpp
index f2fcce0..9d97b42 100644
--- a/src/functions/functions.hpp
+++ b/src/functions/functions.hpp
@@ -14,7 +14,6 @@
    limitations under the License.
 */
 
-#define NEED_BASIC_FUNCTION_CLASSES
 #include <libpmt/lib.hpp>
 
 namespace PartitionManager {