diff --git a/Setup.md b/Setup.md
index 70d2500..9f56071 100644
--- a/Setup.md
+++ b/Setup.md
@@ -6,11 +6,8 @@ TLDR:
 docker compose --profile download up --build
 # wait until its done, then:
 docker compose --profile [ui] up --build
-# where [ui] is one of: auto | auto-cpu | hlky
-```
-<!--
 # where [ui] is one of: auto | auto-cpu | hlky | lstein
--->
+```
 
 Then access from http://localhost:7860/
 
@@ -32,9 +29,8 @@ Next, choose which UI you want to run (you can easily change later):
 - `hlky`: The most popular fork, [Repo by hlky](https://github.com/sd-webui/stable-diffusion-webui)
 - `auto`: Another great fork, many features with neat UI, [Repo by AUTOMATIC1111](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
 - `auto-cpu`: for users without a GPU.
-<!--
-- `lstein`: One of the earliest forks, the UI is simple but has a lot of potential, [Repo by lstein](https://github.com/lstein/stable-diffusion)
--->
+- `lstein`: One of the earliest forks, the UI is simple but has a lot of potential, [Repo by lstein](https://github.com/invoke-ai/InvokeAI)
+
 
 After the download is done, you can run the UI using:
 ```bash