diff --git a/README.md b/README.md
index 6cd1e30..c175241 100644
--- a/README.md
+++ b/README.md
@@ -37,7 +37,7 @@ like [Numpy](https://numpy.org) and [Pillow](https://pypi.org/project/pillow/) f
    ```
 ### Generating an image
 
-Run the provided [main.py](src/mflux/main.py) by specifying a prompt and some optional arguments like so using the `schnell` model:
+Run the provided [main.py](src/mflux/run.py) by specifying a prompt and some optional arguments like so using the `schnell` model:
 
 ```
 python main.py --model schnell --prompt "Luxury food photograph" --steps 2 --seed 2 -q 8
@@ -116,7 +116,7 @@ image = flux.generate_image(
 image.save(path="image.png")
 ```
 
-For more options on how to configure MFLUX, please see [main.py](src/mflux/main.py).
+For more options on how to configure MFLUX, please see [main.py](src/mflux/run.py).
 
 ### Image generation speed (updated)
 
diff --git a/src/mflux/main.py b/src/mflux/run.py
similarity index 100%
rename from src/mflux/main.py
rename to src/mflux/run.py