fix list.get brainfart

shotwn · Apr 5, 2020 · 333887a · 333887a
1 parent 07fb651
commit 333887a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/main.py b/main.py
@@ -89,7 +89,11 @@ def connect_pyuipc(self):
         sim_offset = self.create_offset_set({'SIM_VERSION': [0x3308, 'b']})
         result = sim_offset.read()
         self.pyuipc_open = result['SIM_VERSION']
-        self.opened_sim = SIMULATORS.get(self.pyuipc_open, 'SIM_VERSION', f'Unknown Sim: {self.pyuipc_open}')
+        try:
+            self.opened_sim = SIMULATORS[self.pyuipc_open]
+        except IndexError:
+            self.opened_sim = f'Unknown Sim: {self.pyuipc_open}'
+
         return True
 
     def close_pyuipc(self):