diff --git a/etcd3_wrapper/etcd3_wrapper.py b/etcd3_wrapper/etcd3_wrapper.py
index f105c71..fa9f285 100755
--- a/etcd3_wrapper/etcd3_wrapper.py
+++ b/etcd3_wrapper/etcd3_wrapper.py
@@ -10,9 +10,6 @@ PseudoEtcdMeta = namedtuple("PseudoEtcdMeta", ["key"])
 
 
 class EtcdEntry:
-    # key: str
-    # value: str
-    
     def __init__(self, meta, value, value_in_json=False):
         self.key = meta.key.decode("utf-8")
         self.value = value.decode("utf-8")
@@ -25,7 +22,7 @@ class Etcd3Wrapper:
     def __init__(self, *args, **kwargs):
         self.client = etcd3.client(*args, **kwargs)
     
-    def get(self, *args, value_in_json=False, **kwargs):   
+    def get(self, *args, value_in_json=False, **kwargs) -> Union[EtcdEntry, None]:
         _value, _key = self.client.get(*args, **kwargs)
         if _key is None or _value is None:
             return None