Fix nix-shell setup