diff --git a/process/stats.cpp b/process/stats.cpp
index d6fb23da..b86a8690 100644
--- a/process/stats.cpp
+++ b/process/stats.cpp
@@ -82,21 +82,21 @@ ProcessPlugin *StatsPlugin::copy()
    return new StatsPlugin(*this);
 }
 
-int StatsPlugin::post_create(Flow &rec, const Packet &pkt)
+ProcessPlugin::FlowAction StatsPlugin::post_create(Flow &rec, const Packet &pkt)
 {
    m_packets += 1;
    m_new_flows += 1;
    m_flows_in_cache += 1;
    check_timestamp(pkt);
-   return 0;
+   return ProcessPlugin::FlowAction::GET_ALL_DATA;
 }
 
-int StatsPlugin::post_update(Flow &rec, const Packet &pkt)
+ProcessPlugin::FlowAction StatsPlugin::post_update(Flow &rec, const Packet &pkt)
 {
    m_packets += 1;
    m_cache_hits += 1;
    check_timestamp(pkt);
-   return 0;
+   return ProcessPlugin::FlowAction::GET_ALL_DATA;
 }
 
 void StatsPlugin::pre_export(Flow &rec)
diff --git a/process/stats.hpp b/process/stats.hpp
index b3ff2882..3d884380 100644
--- a/process/stats.hpp
+++ b/process/stats.hpp
@@ -71,8 +71,8 @@ class StatsPlugin : public ProcessPlugin
    std::string get_name() const { return "stats"; }
    ProcessPlugin *copy();
 
-   int post_create(Flow &rec, const Packet &pkt);
-   int post_update(Flow &rec, const Packet &pkt);
+   ProcessPlugin::FlowAction post_create(Flow &rec, const Packet &pkt);
+   ProcessPlugin::FlowAction post_update(Flow &rec, const Packet &pkt);
    void pre_export(Flow &rec);
    void finish(bool print_stats);