diff --git a/tests/unit/ensemble/MajorityVote_unittest.cc b/tests/unit/ensemble/MajorityVote_unittest.cc
index ce32cd3a44b..3e9da8cf933 100644
--- a/tests/unit/ensemble/MajorityVote_unittest.cc
+++ b/tests/unit/ensemble/MajorityVote_unittest.cc
@@ -1,7 +1,7 @@
 #include <shogun/ensemble/MajorityVote.h>
 #include <shogun/lib/SGMatrix.h>
 #include <shogun/lib/SGVector.h>
-#include <shogun/labels/Labels.h>
+#include <shogun/mathematics/Math.h>
 #include <gtest/gtest.h>
 
 using namespace shogun;
diff --git a/tests/unit/ensemble/WeightedMajorityVote_unittest.cc b/tests/unit/ensemble/WeightedMajorityVote_unittest.cc
index 6233dbde758..25601a9b1e1 100644
--- a/tests/unit/ensemble/WeightedMajorityVote_unittest.cc
+++ b/tests/unit/ensemble/WeightedMajorityVote_unittest.cc
@@ -1,7 +1,7 @@
 #include <shogun/ensemble/WeightedMajorityVote.h>
 #include <shogun/lib/SGMatrix.h>
 #include <shogun/lib/SGVector.h>
-#include <shogun/labels/Labels.h>
+#include <shogun/mathematics/Math.h>
 #include <gtest/gtest.h>
 
 using namespace shogun;