@@ -102,7 +102,7 @@ main = bracket (return ()) (\() -> hFlush stdout >> hFlush stderr) $ \() -> do
102102
103103 shell :: Doc
104104 shell = vcat
105− [ text "{ nixpkgs ? import <nixpkgs> {}, compiler ? \"ghc7101\" }:"
105+ [ text "{ nixpkgs ? import <nixpkgs> {}, compiler ? \"default\" }:"
106106 , text ""
107107 , text "let"
108108 , text ""
@@ -110,7 +110,11 @@ main = bracket (return ()) (\() -> hFlush stdout >> hFlush stderr) $ \() -> do
110110 , text ""
111111 , hcat [ text " f = ", deriv', semi ]
112112 , text ""
113− , text " drv = pkgs.haskell.packages.${compiler}.callPackage f {};"
113+ , text " haskellPackages = if compiler == \"default\""
114+ , text " then pkgs.haskellPackages"
115+ , text " else pkgs.haskell.packages.${compiler};"
116+ , text ""
117+ , text " drv = haskellPackages.callPackage f {};"
114118 , text ""
115119 , text "in"
116120 , text ""