diff --git a/prettymaps/app.py b/prettymaps/app.py
index a6a3c1c..85f8f71 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
-from draw import plot, presets
+import prettymaps
 
 logging.basicConfig(
     level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
 )
 
-presets = presets().to_dict()
+presets = prettymaps.presets().to_dict()
 
 # Set the title of the app
 st.title("prettymaps")