diff --git a/rapl_formula/rapl_formula_actor.py b/rapl_formula/rapl_formula_actor.py
index 8e8277e..bc31148 100644
--- a/rapl_formula/rapl_formula_actor.py
+++ b/rapl_formula/rapl_formula_actor.py
@@ -34,7 +34,7 @@
 from functools import reduce
 
 from powerapi.actor import Actor, SocketInterface
-from powerapi.formula import FormulaActor, FormulaState
+from powerapi.formula import FormulaActor, FormulaState, FormulaPoisonPillMessageHandler
 from powerapi.handler import PoisonPillMessageHandler
 from powerapi.message import PoisonPillMessage
 from powerapi.report import HWPCReport
@@ -81,5 +81,5 @@ def setup(self):
         Setup the RAPL formula.
         """
         FormulaActor.setup(self)
-        self.add_handler(PoisonPillMessage, PoisonPillMessageHandler(self.state))
+        self.add_handler(PoisonPillMessage, FormulaPoisonPillMessageHandler(self.state))
         self.add_handler(HWPCReport, RAPLHandler(self.state))