diff --git a/view/domain.py b/view/domain.py
index 992eeff..c2f6ffc 100644
--- a/view/domain.py
+++ b/view/domain.py
@@ -178,7 +178,7 @@ def render(Hscan, Mmixed):
     if len(Hscan) <= 0 or len(Mmixed) <= 0:
         return
     with _lock:
-        with plt.style.context(["science", "nature"]):
+        with plt.style.context(["dark_background"]):
             w, h = plt.figaspect(1 / 3)
             fig, ax = plt.subplots(
                 1, 3, dpi=400, figsize=(w, h), sharex=True, sharey=True
diff --git a/view/requirements.txt b/view/requirements.txt
index 09cd4d5..8f8b9dc 100644
--- a/view/requirements.txt
+++ b/view/requirements.txt
@@ -1,9 +1,10 @@
 git+https://github.com/huawei-noah/HEBO.git#subdirectory=HEBO
-git+https://github.com/lemurpwned/cmtj.git@cmtj-updates
+git+https://github.com/lemurpwned/cmtj.git@streamlit/ui
 matplotlib
 networkx
 numba
 numpy
 scipy
+setuptools
 streamlit
 tqdm