diff --git a/src/daal4py.cpp b/src/daal4py.cpp
index bbb37373b8..df8768f0d5 100755
--- a/src/daal4py.cpp
+++ b/src/daal4py.cpp
@@ -856,7 +856,7 @@ void c_train_test_split(data_or_file & orig, data_or_file & train, data_or_file
 
 double c_roc_auc_score(data_or_file & y_true, data_or_file & y_test)
 {
-#if __INTEL_DAAL__ == 2021 && INTEL_DAAL_VERSION >= 20210200
+#if __INTEL_DAAL__ >= 2021 && INTEL_DAAL_VERSION >= 20210200
     const size_t col_true = y_true.table->getNumberOfColumns();
     const size_t row_true = y_true.table->getNumberOfRows();
     const size_t col_test = y_test.table->getNumberOfColumns();
diff --git a/src/daal4py.h b/src/daal4py.h
index a4d792a1a3..560f98e7e4 100755
--- a/src/daal4py.h
+++ b/src/daal4py.h
@@ -52,7 +52,7 @@ using daal::services::LibraryVersionInfo;
 #include "data_management/data/internal/finiteness_checker.h"
 #include "data_management/data/internal/train_test_split.h"
 
-#if __INTEL_DAAL__ == 2021 && INTEL_DAAL_VERSION >= 20210200
+#if __INTEL_DAAL__ >= 2021 && INTEL_DAAL_VERSION >= 20210200
     #include "data_management/data/internal/roc_auc_score.h"
 #endif
 #if __INTEL_DAAL__ >= 2021 && INTEL_DAAL_VERSION >= 20210600