diff --git a/src/clj/auto_ap/jobs/square.clj b/src/clj/auto_ap/jobs/square.clj
index b9a18f1f..14dbd7c0 100644
--- a/src/clj/auto_ap/jobs/square.clj
+++ b/src/clj/auto_ap/jobs/square.clj
@@ -3,9 +3,15 @@
   (:require [auto-ap.square.core :as square]
             [auto-ap.utils :refer [heartbeat]]
             [mount.core :as mount]
-            [auto-ap.datomic :refer [conn]]))
+            [auto-ap.datomic :refer [conn]]
+            [clojure.tools.logging :as log]
+            [unilog.context :as lc]))
 
 (defn -main [& _]
-  (mount/start (mount/only #{#'conn}))
-  ((heartbeat square/upsert-all "square-loading")))
+  (lc/with-context {:background-job "square-loading"}
+    (mount/start (mount/only #{#'conn}))
+    ((heartbeat square/upsert-all "square-loading"))
+    (mount/stop (mount/only #{#'conn}))
+    (log/info "Stopping Square loading")
+    (System/exit 0)))