diff --git a/python-data-science.nix b/python-data-science.nix
index 0472ad7..345cf04 100644
--- a/python-data-science.nix
+++ b/python-data-science.nix
@@ -1,8 +1,8 @@
-{ pkgs ? import <nixpkgs> { } }:
+{ config, lib, pkgs, ... }:
 
 with pkgs;
 
-mkShell {
+{
   buildInputs = [
     python3
     python3Packages.jupyter
@@ -12,8 +12,4 @@ mkShell {
     python3Packages.matplotlib
     python3Packages.seaborn
   ];
-
-  shellHook = ''
-    jupyter notebook
-  '';
 }