diff --git a/src/charm.py b/src/charm.py
index 5c0c649..f950b69 100755
--- a/src/charm.py
+++ b/src/charm.py
@@ -32,10 +32,10 @@
     ModelError,
     RelationBrokenEvent,
     WaitingStatus,
+    main,
 )
 from ops.charm import CharmBase
 from ops.framework import EventBase
-from ops.main import main
 from ops.pebble import Layer
 
 logger = logging.getLogger(__name__)