From 25ef8cdae5c39135121c6b617fea747315dea3ff Mon Sep 17 00:00:00 2001
From: Paul Oliver <puzza007@gmail.com>
Date: Fri, 21 Aug 2015 10:55:06 +0200
Subject: [PATCH] Use module() instead of atom() where applicable

---
 src/beam_stats_config.erl   | 2 +-
 src/beam_stats_consumer.erl | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/beam_stats_config.erl b/src/beam_stats_config.erl
index 45d0fc9..a6d5d6d 100644
--- a/src/beam_stats_config.erl
+++ b/src/beam_stats_config.erl
@@ -17,7 +17,7 @@ production_interval() ->
     get_env(production_interval).
 
 -spec consumers() ->
-    [{ConsumerModule :: atom(), ConsumerDefinedOptionsData :: term()}].
+    [{ConsumerModule :: module(), ConsumerDefinedOptionsData :: term()}].
 consumers() ->
     get_env(consumers).
 
diff --git a/src/beam_stats_consumer.erl b/src/beam_stats_consumer.erl
index fa06695..eb8cbb7 100644
--- a/src/beam_stats_consumer.erl
+++ b/src/beam_stats_consumer.erl
@@ -52,7 +52,7 @@
 -define(SIGNAL_CONSUMPTION , beam_stats_consumption_signal).
 
 -record(state,
-    { consumer_module      :: atom()
+    { consumer_module      :: module()
     , consumer_state       :: term()
     , consumption_interval :: non_neg_integer()
     , beam_stats_queue     :: queue()
@@ -62,7 +62,7 @@
 %%  Public API
 %% ============================================================================
 
--spec add(atom(), term()) ->
+-spec add(module(), term()) ->
     supervisor:startchild_ret().
 add(ConsumerModule, ConsumerOptions) ->
     beam_stats_sup_consumers:start_child(ConsumerModule, ConsumerOptions).
-- 
2.20.1