diff --git a/main.go b/main.go
index a7fab7c..19d5ddc 100644
--- a/main.go
+++ b/main.go
@@ -43,6 +43,11 @@ func main() {
 
 		pprof.StartCPUProfile(f)
 		log.V(1).Info("started profiling")
+
+		defer func() {
+			pprof.StopCPUProfile()
+			log.V(1).Info("stopped profiling")
+		}()
 	}
 
 	// Load the config file.
@@ -65,11 +70,6 @@ func main() {
 		<-interrupts
 		log.V(1).Info("caught interrupt, shutting down...")
 
-		if profile {
-			pprof.StopCPUProfile()
-			log.V(1).Info("stopped profiling")
-		}
-
 		err := s.Stop()
 		if err != nil {
 			log.Fatalf("failed to shutdown cleanly: %s", err)