4 files changed, 8 insertions, 6 deletions

diff --git a/lambdacube-compiler.cabal b/lambdacube-compiler.cabal
index 5e667c30..67054501 100644
--- a/lambdacube-compiler.cabal
+++ b/lambdacube-compiler.cabal
@@ -39,12 +39,12 @@ source-repository head
 library
   exposed-modules:
     -- Compiler
+    LambdaCube.Compiler
     LambdaCube.Compiler.Pretty
     LambdaCube.Compiler.Lexer
     LambdaCube.Compiler.Parser
     LambdaCube.Compiler.Infer
     LambdaCube.Compiler.CoreToIR
-    LambdaCube.Compiler.Driver
   other-extensions:
     LambdaCase
     PatternSynonyms
diff --git a/src/LambdaCube/Compiler/Driver.hs b/src/LambdaCube/Compiler.hs
index ef856112..d8b09431 100644
--- a/src/LambdaCube/Compiler/Driver.hs
+++ b/src/LambdaCube/Compiler.hs
@@ -6,7 +6,7 @@
 {-# LANGUAGE FlexibleContexts #-}
 {-# LANGUAGE ScopedTypeVariables #-}
 {-# OPTIONS_GHC -fno-warn-orphans #-}  -- instance MonadMask m => MonadMask (ExceptT e m)
-module LambdaCube.Compiler.Driver
+module LambdaCube.Compiler
     ( Backend(..)
     , Pipeline
     , Infos, listInfos, Range(..)
@@ -20,6 +20,9 @@ module LambdaCube.Compiler.Driver
     , ioFetch
     , getDef, compileMain, preCompile
     , removeFromCache
+
+    , compilePipeline
+    , ppShow
     ) where
 
 import Data.List
diff --git a/test/runTests.hs b/test/runTests.hs
index 88c2f89f..c9375d5d 100644
--- a/test/runTests.hs
+++ b/test/runTests.hs
@@ -24,9 +24,7 @@ import qualified Data.Text as T
 import qualified Data.Text.IO as TIO
 import Text.Printf
 
-import LambdaCube.Compiler.Pretty (ppShow)
-import LambdaCube.Compiler.CoreToIR (compilePipeline)
-import LambdaCube.Compiler.Driver
+import LambdaCube.Compiler
 
 ------------------------------------------ utils
 
diff --git a/tool/Compiler.hs b/tool/Compiler.hs
index aa5a0e60..0723c0c8 100644
--- a/tool/Compiler.hs
+++ b/tool/Compiler.hs
@@ -2,9 +2,10 @@
 import Options.Applicative
 import Data.Aeson
 import qualified Data.ByteString.Lazy as B
-import LambdaCube.Compiler.Driver
 import System.FilePath
+
 import Paths_lambdacube_compiler (getDataDir)
+import LambdaCube.Compiler
 
 data Config
   = Config