1 files changed, 18 insertions, 4 deletions
diff --git a/tests/link/driver.cxx b/tests/link/driver.cxx
index a5e94bb..352cadd 100644
--- a/tests/link/driver.cxx
+++ b/tests/link/driver.cxx
@@ -2,14 +2,28 @@
 // copyright : Copyright (c) 2014-2017 Code Synthesis Ltd
 // license   : MIT; see accompanying LICENSE file
 
-#include <set>
 #include <cassert>
+
+#ifndef __cpp_lib_modules
+#include <set>
 #include <utility>      // pair
 #include <system_error>
+#endif
 
-#include <libbutl/path.hxx>
-#include <libbutl/fdstream.hxx>
-#include <libbutl/filesystem.hxx>
+// Other includes.
+
+#ifdef __cpp_modules
+#ifdef __cpp_lib_modules
+import std.core;
+#endif
+import butl.path;
+import butl.fdstream;
+import butl.filesystem;
+#else
+#include <libbutl/path.mxx>
+#include <libbutl/fdstream.mxx>
+#include <libbutl/filesystem.mxx>
+#endif
 
 using namespace std;
 using namespace butl;