X-Git-Url: https://git.xandkar.net/?p=beam_stats.git;a=blobdiff_plain;f=src%2Fbeam_stats_consumer.erl;h=eb8cbb7d2094b8ae8dca88c748ceca48a0f52295;hp=fa0669517245635e03739416a22cad983aeffb1e;hb=25ef8cdae5c39135121c6b617fea747315dea3ff;hpb=1293e92b54ae6989738e45fe6dd3d837a904edaa

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).