diff --git a/src/TensorFlow/OpGen.hs b/src/TensorFlow/OpGen.hs
--- a/src/TensorFlow/OpGen.hs
+++ b/src/TensorFlow/OpGen.hs
@@ -51,7 +51,8 @@
 
 import Data.Foldable (toList)
 import Data.Maybe (fromMaybe)
-import Data.ProtoLens (def, showMessage)
+import Data.ProtoLens.Default(def)
+import Data.ProtoLens (showMessage)
 import Data.List (sortOn)
 import Data.List.NonEmpty (NonEmpty)
 import qualified Data.List.NonEmpty as NE
@@ -60,7 +61,9 @@
 import Proto.Tensorflow.Core.Framework.OpDef
   ( OpList
   , OpDef
-  , attr
+  )
+import Proto.Tensorflow.Core.Framework.OpDef_Fields
+  ( attr
   , inputArg
   , name
   , op
@@ -69,9 +72,9 @@
 import Proto.Tensorflow.Core.Framework.Types (DataType(..))
 import System.FilePath (takeBaseName)
 import TensorFlow.OpGen.ParsedOp
+import Data.Semigroup ((<>))
 import Text.PrettyPrint.Mainland
   ( Doc
-  , (<>)
   , (<+>)
   , (</>)
   , (<+/>)
diff --git a/src/TensorFlow/OpGen/ParsedOp.hs b/src/TensorFlow/OpGen/ParsedOp.hs
--- a/src/TensorFlow/OpGen/ParsedOp.hs
+++ b/src/TensorFlow/OpGen/ParsedOp.hs
@@ -31,12 +31,14 @@
 import Data.Text (Text)
 import qualified Data.Text as Text
 import Lens.Family2 ((^.))
-import Proto.Tensorflow.Core.Framework.AttrValue (list)
+import Proto.Tensorflow.Core.Framework.AttrValue_Fields (list)
 import Proto.Tensorflow.Core.Framework.OpDef
     ( OpDef
     , OpDef'ArgDef
     , OpDef'AttrDef
-    , allowedValues
+    )
+import Proto.Tensorflow.Core.Framework.OpDef_Fields
+    ( allowedValues
     , attr
     , maybe'defaultValue
     , description
@@ -51,6 +53,7 @@
     , typeAttr
     , type'
     )
+
 import Proto.Tensorflow.Core.Framework.Types (DataType(DT_RESOURCE))
 
 data ParsedOp = ParsedOp
diff --git a/tensorflow-opgen.cabal b/tensorflow-opgen.cabal
--- a/tensorflow-opgen.cabal
+++ b/tensorflow-opgen.cabal
@@ -1,5 +1,5 @@
 name:                tensorflow-opgen
-version:             0.2.0.0
+version:             0.2.0.1
 synopsis:            Code generation for TensorFlow operations.
 description:         Please see README.md
 homepage:            https://github.com/tensorflow/haskell#readme
@@ -16,7 +16,7 @@
   hs-source-dirs:      src
   exposed-modules:     TensorFlow.OpGen.ParsedOp
                      , TensorFlow.OpGen
-  build-depends:  proto-lens == 0.2.*
+  build-depends:  proto-lens >= 0.4.0 && < 0.6.0
                 , tensorflow-proto == 0.2.*
                 , base >= 4.7 && < 5
                 , bytestring