diff --git a/onedal/primitives/finiteness_checker.cpp b/onedal/primitives/finiteness_checker.cpp
index 51a3ef161a..761ee28de9 100644
--- a/onedal/primitives/finiteness_checker.cpp
+++ b/onedal/primitives/finiteness_checker.cpp
@@ -14,7 +14,12 @@
 * limitations under the License.
 *******************************************************************************/
 
+// fix error with missing headers
+#if defined(ONEDAL_VERSION) && ONEDAL_VERSION >= 20250200
+#include "oneapi/dal/algo/finiteness_checker.hpp
+#else
 #include "oneapi/dal/algo/finiteness_checker/compute.hpp"
+#endif // defined(ONEDAL_VERSION) && ONEDAL_VERSION >= 20250200
 
 #include "onedal/common.hpp"
 #include "onedal/version.hpp"