       (is (= "first-metric" (-> @events first :metric-name)))
 
       ;; Restart
-      (sut/stop thread-pool)
-      (sut/init! tc/zookeeper-config)
+      (consume/stop thread-pool)
+      (consume/init! tc/zookeeper-config)
       (reset! events [])
 
       ;; Second Run
-      (let [thread-pool (sut/start {topic 1} process-fn)]
+      (let [thread-pool (consume/start {topic 1} process-fn)]
         (is (client/gauge :second-metric nil 5))
 
         (tu/wait-for #(= 1 (count @events)) :interval 0.5 :timeout 2)

(ns wonko.kafka.consume-test
  (:require [clojure.test :refer :all]
            [wonko.kafka.consume :as sut]
            [wonko.test-utils :as tu]
            [wonko-client.core :as client]
            [wonko.test-config :as tc]

        (is (= "gauge" (-> @events first :metric-type)))
        (is (= "second-metric" (-> @events first :metric-name)))

        (consume/stop thread-pool)))))
