1 files changed, 17 insertions, 3 deletions
diff --git a/tests/tab-parser/driver.cxx b/tests/tab-parser/driver.cxx
index 238c12a..d5ff5a0 100644
--- a/tests/tab-parser/driver.cxx
+++ b/tests/tab-parser/driver.cxx
@@ -2,12 +2,26 @@
 // copyright : Copyright (c) 2014-2017 Code Synthesis Ltd
 // license   : MIT; see accompanying LICENSE file
 
-#include <ios>      // ios::failbit, ios::badbit
 #include <cassert>
+
+#ifndef __cpp_lib_modules
+#include <string>
 #include <iostream>
+#endif
+
+// Other includes.
 
-#include <libbutl/utility.hxx>    // operator<<(ostream,exception)
-#include <libbutl/tab-parser.hxx>
+#ifdef __cpp_modules
+#ifdef __cpp_lib_modules
+import std.core;
+import std.io;
+#endif
+import butl.utility;    // operator<<(ostream,exception)
+import butl.tab_parser;
+#else
+#include <libbutl/utility.mxx>
+#include <libbutl/tab-parser.mxx>
+#endif
 
 using namespace std;
 using namespace butl;