diff --git a/nix/orbax-checkpoint.nix b/nix/orbax-checkpoint.nix
index 9742338..192172d 100644
--- a/nix/orbax-checkpoint.nix
+++ b/nix/orbax-checkpoint.nix
@@ -44,5 +44,5 @@ buildPythonPackage rec {
     description = "Checkpointing library for JAX-based models";
     license = licenses.asl20;
     homepage = "https://github.com/google/orbax";
-  };
+  };  
 }
diff --git a/nix/tensorstore.nix b/nix/tensorstore.nix
index 4ba841f..3455819 100644
--- a/nix/tensorstore.nix
+++ b/nix/tensorstore.nix
@@ -1,30 +1,61 @@
+# { buildPythonPackage
+# , fetchFromGitHub
+# , setuptools
+# , pkgs
+# }:
+
+# buildPythonPackage rec {
+#   name = "tensorstore";
+  
+#   src = fetchFromGitHub {
+#     owner = "google";
+#     repo = "tensorstore";
+#     rev = "v0.1.60";
+#     hash = "sha256-rT0R1x51xHAElPwernUjBIIneRhncnsohMRAIhXyaYk=";
+#   };
+#   format = "pyproject";
+
+#   buildinputs = [
+#     pkgs.bazel
+#   ];
+  
+#   propagatedBuildInputs = [
+#     setuptools
+#   ];
+
+#   preConfigure = ''
+#     export HOME=$PWD
+#   '';
+
+# }
+
 { buildPythonPackage
-, fetchFromGitHub
-, setuptools
+, fetchPypi
 , pkgs
+, pip
+, lib
+, setuptools
 }:
 
 buildPythonPackage rec {
-  name = "tensorstore";
-  
-  src = fetchFromGitHub {
-    owner = "google";
-    repo = "tensorstore";
-    rev = "v0.1.60";
-    hash = "sha256-rT0R1x51xHAElPwernUjBIIneRhncnsohMRAIhXyaYk=";
+  pname = "tensorstore";
+  version = "0.1.6";
+
+  src = fetchPypi {
+    inherit pname version;
+    sha256 = "sha256-MtzIVk6oCmF0gSFisrruxSUyNdVwIciMZNY0sORN1Jg=";
   };
-  format = "pyproject";
 
-  buildinputs = [
-    pkgs.bazel
-  ];
-  
   propagatedBuildInputs = [
+    pip
     setuptools
   ];
 
-  preConfigure = ''
-    export HOME=$PWD
-  '';
-  
+  # nativeBuildInputs = [pkgs.python3Packages.pip];
+
+  meta = with lib; {
+    description = "TensorStore";
+    license = licenses.asl20;
+    homepage = "https://github.com/google/tensorstore";
+  };  
 }