diff --git a/u7/hash_table.py b/u7/hash_table.py
index 56c5502..fd53964 100644
--- a/u7/hash_table.py
+++ b/u7/hash_table.py
@@ -1,6 +1,4 @@
 """Hash tabuľka s lineárnym sondovaním a reťazcovými kľúčmi."""
-from operator import index
-from unittest import result
 
 
 class HashTable:
@@ -75,11 +73,10 @@ class HashTable:
             self.size -= 1
 
     def __contains__(self, key):
-        try:
-            self.probe(key)
-            return True
-        except KeyError:
+        if self.table[self.probe(key)] is self.empty or self.table[self.probe(key)] is self.deleted:
             return False
+        else:
+            return True
 
     def __len__(self):
         return self.size