diff --git a/userspace/engine/token_bucket.cpp b/userspace/engine/token_bucket.cpp
index f40112e7..1c30c649 100644
--- a/userspace/engine/token_bucket.cpp
+++ b/userspace/engine/token_bucket.cpp
@@ -73,3 +73,8 @@ bool token_bucket::claim(uint64_t now)
 
 	return true;
 }
+
+double token_bucket::get_tokens()
+{
+	return m_tokens;
+}
diff --git a/userspace/engine/token_bucket.h b/userspace/engine/token_bucket.h
index b60b5319..2f767ef4 100644
--- a/userspace/engine/token_bucket.h
+++ b/userspace/engine/token_bucket.h
@@ -38,6 +38,10 @@ public:
 	// internal metrics.
 	//
 	bool claim(uint64_t now = 0);
+
+	// Return the current number of tokens available
+	double get_tokens();
+
 private:
 
 	//