diff --git a/whereDecision.py b/whereDecision.py
index 786ecd2..1962e3a 100644
--- a/whereDecision.py
+++ b/whereDecision.py
@@ -51,6 +51,8 @@ class WhereDecision:
                 radioaktywne.append(0)
                 niebezpieczne.append(0)
                 lokacja.append(str("("+str(regal[0])+", "+str(regal[1])+")"))
+        if len(zwykle) == 0:
+            return [1, 2]
         z = list(zip(zwykle, kruche, latwopalne, radioaktywne, niebezpieczne))
         prenumeratorzy = pd.DataFrame({"lokacja": lokacja})
         prenumeratorzy["lokacja"], lokacja_kody = pd.factorize(prenumeratorzy["lokacja"])