diff --git a/koboldai_settings.py b/koboldai_settings.py
index 5b2ab667..9c433e61 100644
--- a/koboldai_settings.py
+++ b/koboldai_settings.py
@@ -243,7 +243,8 @@ class model_settings(settings):
             else:
                 self.tqdm.update(1)
                 self.tqdm_progress = int(float(self.generated_tkns)/float(self.genamt)*100)
-                self.tqdm_rem_time = str(datetime.timedelta(seconds=int(float(self.genamt-self.generated_tkns)/self.tqdm.format_dict['rate'])))
+                if self.tqdm.format_dict['rate'] is not None:
+                    self.tqdm_rem_time = str(datetime.timedelta(seconds=int(float(self.genamt-self.generated_tkns)/self.tqdm.format_dict['rate'])))
         #Setup TQDP for model loading
         if name == "loaded_layers" and 'tqdm' in self.__dict__:
             if value == 0:
diff --git a/test_aiserver.py b/test_aiserver.py
index b9bf4606..0959d98c 100644
--- a/test_aiserver.py
+++ b/test_aiserver.py
@@ -13,7 +13,7 @@ def client_data():
     app = aiserver.app
     #app.test_client_class = FlaskLoginClient
     client_conn = app.test_client()
-    socketio_client = aiserver.socketio.test_client(app, flask_test_client=client_conn)
+    socketio_client = aiserver.socketio.test_client(app, flask_test_client=client_conn, query_string="ui=1")
     #Clear out the connection message
     response = socketio_client.get_received()
     return (client_conn, app, socketio_client)