diff --git a/prettymaps/app.py b/prettymaps/app.py
index 85f8f71..a6a3c1c 100644
--- a/prettymaps/app.py
+++ b/prettymaps/app.py
@@ -2,13 +2,13 @@ import streamlit as st
 import logging
 from matplotlib import pyplot as plt
 import subprocess
-import prettymaps
+from draw import plot, presets
 
 logging.basicConfig(
     level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
 )
 
-presets = prettymaps.presets().to_dict()
+presets = presets().to_dict()
 
 # Set the title of the app
 st.title("prettymaps")