diff --git a/hasktorch.cabal b/hasktorch.cabal
--- a/hasktorch.cabal
+++ b/hasktorch.cabal
@@ -1,16 +1,20 @@
+cabal-version:       3.0
 name:                hasktorch
-version:             0.2.1.0
+version:             0.2.1.1
 synopsis:            Haskell bindings to libtorch, supporting both typed and untyped tensors.
 description:         Hasktorch is a library for tensors and neural networks in Haskell. It is an independent open source community project which leverages the core C++ libraries shared by PyTorch.
 homepage:            https://github.com/hasktorch/hasktorch#readme
-license:             BSD3
+license:             BSD-3-Clause
 license-file:        LICENSE
 author:              Hasktorch Contributor Team
 maintainer:          hasktorch@gmail.com
 copyright:           2019 Austin Huang
 category:            Codegen
 build-type:          Custom
-cabal-version:       1.24
+extra-source-files:
+  ./test/data/numpy_rawfile
+  ./test/data/mnist-sample-labels-idx1-ubyte.gz
+  ./test/data/mnist-sample-images-idx3-ubyte.gz
 
 custom-setup
   setup-depends:
@@ -96,7 +100,6 @@
                     , Torch.Index
                     , Torch.Jit
 
- other-modules:       Paths_hasktorch
  hs-source-dirs:      src
  default-language:    Haskell2010
  ghc-options:         -fplugin GHC.TypeLits.Normalise -fplugin GHC.TypeLits.KnownNat.Solver -fplugin GHC.TypeLits.Extra.Solver -fconstraint-solver-iterations=0 -fplugin GHC.NotExport.Plugin
diff --git a/test/Torch/Distributions/BernoulliSpec.hs b/test/Torch/Distributions/BernoulliSpec.hs
--- a/test/Torch/Distributions/BernoulliSpec.hs
+++ b/test/Torch/Distributions/BernoulliSpec.hs
@@ -62,9 +62,9 @@
     toList (Just t) `shouldBe` [True, True]
 
   it "logProb" $ do
-    -- putStrLn . show $ logProb d $ D.asTensor [[0.3, 0.5 :: Float]]
-    let t :: Tnsr 'D.Float '[1, 2] = UnsafeMkTensor $ logProb d $ D.asTensor [[0.3, 0.5 :: Float]]
-    F.allclose (toDynamic t) (D.asTensor [[-0.6749387, -0.7530129 :: Float]]) 0.001 0.001 False `shouldBe` True
+    -- putStrLn . show $ logProb d $ D.asTensor [0.3, 0.5 :: Float]
+    let t :: Tnsr 'D.Float '[2] = UnsafeMkTensor $ logProb d $ D.asTensor [0.3, 0.5 :: Float]
+    F.allclose (toDynamic t) (D.asTensor [-0.6749387, -0.7530129 :: Float]) 0.001 0.001 False `shouldBe` True
 
   it "entropy" $ do
     -- putStrLn . show $ entropy d
diff --git a/test/data/mnist-sample-images-idx3-ubyte.gz b/test/data/mnist-sample-images-idx3-ubyte.gz
new file mode 100644
Binary files /dev/null and b/test/data/mnist-sample-images-idx3-ubyte.gz differ
diff --git a/test/data/mnist-sample-labels-idx1-ubyte.gz b/test/data/mnist-sample-labels-idx1-ubyte.gz
new file mode 100644
Binary files /dev/null and b/test/data/mnist-sample-labels-idx1-ubyte.gz differ
diff --git a/test/data/numpy_rawfile b/test/data/numpy_rawfile
new file mode 100644
Binary files /dev/null and b/test/data/numpy_rawfile differ