diff --git a/dataframe-hasktorch.cabal b/dataframe-hasktorch.cabal
--- a/dataframe-hasktorch.cabal
+++ b/dataframe-hasktorch.cabal
@@ -1,6 +1,6 @@
 cabal-version:      3.0
 name:               dataframe-hasktorch
-version:            0.1.0.2
+version:            0.2.0.0
 synopsis:           Converts between dataframes and hasktorch tensors
 
 description:        
@@ -37,8 +37,13 @@
 
     build-depends:    base >= 4.11 && < 5,
                       vector ^>= 0.13,
-                      dataframe >= 0.3.3.3 && < 0.6,
+                      dataframe-core ^>= 1.0,
+                      dataframe-operations ^>= 1.0,
                       hasktorch >= 0.2.1.6 && < 0.3
+    if impl(ghc >= 9.12)
+      build-depends: ghc-typelits-natnormalise == 0.9.3
+    else
+      build-depends: ghc-typelits-natnormalise >= 0.7.7 && < 0.9
 
     hs-source-dirs:   src
 
diff --git a/src/DataFrame/Hasktorch.hs b/src/DataFrame/Hasktorch.hs
--- a/src/DataFrame/Hasktorch.hs
+++ b/src/DataFrame/Hasktorch.hs
@@ -8,10 +8,10 @@
 import qualified Data.Vector as V
 import qualified Data.Vector.Unboxed as VU
 import qualified Data.Vector.Unboxed.Mutable as VUM
-import qualified DataFrame as D
+import qualified DataFrame.Operations.Core as D
 
 import Control.Exception (throw)
-import DataFrame (DataFrame)
+import DataFrame.Internal.DataFrame (DataFrame)
 import Torch
 
 {- | Converts a dataframe to a floating-point tensor.