diff --git a/src/TensorFlow/Logging.hs b/src/TensorFlow/Logging.hs
--- a/src/TensorFlow/Logging.hs
+++ b/src/TensorFlow/Logging.hs
@@ -60,7 +60,7 @@
 import Data.ByteString (ByteString)
 import Data.Conduit ((.|))
 import Data.Conduit.TQueue (sourceTBMQueue)
-import Data.Default (def)
+import Data.ProtoLens.Default(def)
 import Data.Int (Int64)
 import Data.Word (Word8, Word16)
 import Data.ProtoLens (encodeMessage)
@@ -69,7 +69,8 @@
 import Lens.Family2 ((.~), (&))
 import Network.HostName (getHostName)
 import Proto.Tensorflow.Core.Framework.Summary (Summary)
-import Proto.Tensorflow.Core.Util.Event (Event, fileVersion, graphDef, step, summary, wallTime)
+import Proto.Tensorflow.Core.Util.Event (Event)
+import Proto.Tensorflow.Core.Util.Event_Fields (fileVersion, graphDef, step, summary, wallTime)
 import System.Directory (createDirectoryIfMissing)
 import System.FilePath ((</>))
 import TensorFlow.Build (MonadBuild, Build, asGraphDef)
diff --git a/tensorflow-logging.cabal b/tensorflow-logging.cabal
--- a/tensorflow-logging.cabal
+++ b/tensorflow-logging.cabal
@@ -1,5 +1,5 @@
 name:                tensorflow-logging
-version:             0.2.0.0
+version:             0.2.0.1
 synopsis:            TensorBoard related functionality.
 description:         Please see README.md
 homepage:            https://github.com/tensorflow/haskell#readme
@@ -24,7 +24,7 @@
                 , filepath
                 , hostname
                 , lens-family
-                , proto-lens == 0.2.*
+                , proto-lens >= 0.4.0 && < 0.6.0
                 , resourcet
                 , stm
                 , stm-chans
diff --git a/tests/LoggingTest.hs b/tests/LoggingTest.hs
--- a/tests/LoggingTest.hs
+++ b/tests/LoggingTest.hs
@@ -17,11 +17,12 @@
 
 import Control.Monad.Trans.Resource (runResourceT)
 import Data.Conduit ((.|))
-import Data.Default (def)
+import Data.ProtoLens.Message (defMessage)
 import Data.List ((\\))
 import Data.ProtoLens (encodeMessage, decodeMessageOrDie)
 import Lens.Family2 ((^.), (.~), (&))
-import Proto.Tensorflow.Core.Util.Event (Event, graphDef, fileVersion, step)
+import Proto.Tensorflow.Core.Util.Event (Event)
+import Proto.Tensorflow.Core.Util.Event_Fields (graphDef, fileVersion, step)
 import System.Directory (getDirectoryContents)
 import System.FilePath ((</>))
 import System.IO.Temp (withSystemTempDirectory)
@@ -44,9 +45,9 @@
 testEventWriter = testCase "EventWriter" $
     withSystemTempDirectory "event_writer_logs" $ \dir -> do
         assertEqual "No file before" [] =<< listDirectory dir
-        let expected = [ (def :: Event) & step .~ 10
-                       , def & step .~ 222
-                       , def & step .~ 8
+        let expected = [ (defMessage :: Event) & step .~ 10
+                       , defMessage & step .~ 222
+                       , defMessage & step .~ 8
                        ]
         withEventWriter dir $ \eventWriter ->
             mapM_ (logEvent eventWriter) expected
@@ -65,7 +66,7 @@
     withSystemTempDirectory "event_writer_logs" $ \dir -> do
         let graphBuild = noOp :: Build ControlNode
             expectedGraph = asGraphDef graphBuild
-            expectedGraphEvent = (def :: Event) & graphDef .~ (encodeMessage expectedGraph)
+            expectedGraphEvent = (defMessage :: Event) & graphDef .~ (encodeMessage expectedGraph)
 
         withEventWriter dir $ \eventWriter ->
             logGraph eventWriter graphBuild