diff --git a/userspace/engine/token_bucket.cpp b/userspace/engine/token_bucket.cpp
index c1ae9b0c..93f04924 100644
--- a/userspace/engine/token_bucket.cpp
+++ b/userspace/engine/token_bucket.cpp
@@ -19,6 +19,7 @@ along with falco.  If not, see <http://www.gnu.org/licenses/>.
 #include <cstddef>
 #include <sys/time.h>
 
+#include "utils.h"
 #include "token_bucket.h"
 
 token_bucket::token_bucket()
@@ -35,14 +36,14 @@ void token_bucket::init(uint32_t rate, uint32_t max_tokens)
 	m_rate = rate;
 	m_max_tokens = max_tokens;
 	m_tokens = max_tokens;
-	m_last_seen = get_epoch_ns();
+	m_last_seen = sinsp_utils::get_current_time_ns();
 }
 
 bool token_bucket::claim()
 {
 	// Determine the number of tokens gained. Delta between
 	// last_seen and now, divided by the rate.
-	uint64_t now = get_epoch_ns();
+	uint64_t now = sinsp_utils::get_current_time_ns();
 	uint64_t tokens_gained = (now - m_last_seen) / (m_rate * 1000000000);
 	m_last_seen = now;
 
@@ -68,11 +69,3 @@ bool token_bucket::claim()
 
 	return true;
 }
-
-uint64_t token_bucket::get_epoch_ns()
-{
-    struct timeval tv;
-    gettimeofday(&tv, NULL);
-
-    return tv.tv_sec * (uint64_t) 1000000000 + (tv.tv_usec * 1000);
-}
diff --git a/userspace/engine/token_bucket.h b/userspace/engine/token_bucket.h
index f9f62e00..b9c0d8b9 100644
--- a/userspace/engine/token_bucket.h
+++ b/userspace/engine/token_bucket.h
@@ -40,9 +40,6 @@ public:
 	bool claim();
 private:
 
-	// Utility function to get the time in nanoseconds since the epoch.
-	uint64_t get_epoch_ns();
-
 	//
 	// The number of tokens generated per second.
 	//