diff --git a/flake.nix b/flake.nix
index b0c9ff8..159d416 100644
--- a/flake.nix
+++ b/flake.nix
@@ -5,14 +5,20 @@
       pkgs = artiq.inputs.nixpkgs.legacyPackages.x86_64-linux;
       aqmain = artiq.packages.x86_64-linux;
     in {
-      defaultPackage.x86_64-linux = pkgs.buildEnv {
+      devShells.x86_64-linux.default = pkgs.mkShell {
         name = "aq6140-env";
-        paths = [
+        buildInputs = [
           (pkgs.python3.withPackages(ps: [
             aqmain.artiq
             ps.pyvisa ps.pyvisa-py ps.numpy
+            ps.pyqtgraph
           ]))
         ];
+        shellHook=
+          ''
+          export QT_PLUGIN_PATH=${pkgs.qt5.qtbase}/${pkgs.qt5.qtbase.dev.qtPluginPrefix}
+          export QML2_IMPORT_PATH=${pkgs.qt5.qtbase}/${pkgs.qt5.qtbase.dev.qtQmlPrefix}
+          '';
       };
     };
 }