From ba917ec5d64ebf80a73193672e666789eea66b24 Mon Sep 17 00:00:00 2001
From: Joran Honig <joran.honig@gmail.com>
Date: Thu, 28 Mar 2019 13:41:32 +0100
Subject: [PATCH] make the existing plugins explicitly extend LaserPlugin

---
 mythril/laser/ethereum/plugins/implementations/benchmark.py    | 3 ++-
 .../laser/ethereum/plugins/implementations/mutation_pruner.py  | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/mythril/laser/ethereum/plugins/implementations/benchmark.py b/mythril/laser/ethereum/plugins/implementations/benchmark.py
index 530c312e..4a91be76 100644
--- a/mythril/laser/ethereum/plugins/implementations/benchmark.py
+++ b/mythril/laser/ethereum/plugins/implementations/benchmark.py
@@ -1,4 +1,5 @@
 from mythril.laser.ethereum.svm import LaserEVM
+from mythril.laser.ethereum.plugins import LaserPlugin
 from time import time
 import matplotlib.pyplot as plt
 import logging
@@ -6,7 +7,7 @@ import logging
 log = logging.getLogger(__name__)
 
 
-class BenchmarkPlugin:
+class BenchmarkPlugin(LaserPlugin):
     """Benchmark Plugin
 
     This plugin aggregates the following information:
diff --git a/mythril/laser/ethereum/plugins/implementations/mutation_pruner.py b/mythril/laser/ethereum/plugins/implementations/mutation_pruner.py
index 38ad1410..0e0ff62b 100644
--- a/mythril/laser/ethereum/plugins/implementations/mutation_pruner.py
+++ b/mythril/laser/ethereum/plugins/implementations/mutation_pruner.py
@@ -1,6 +1,7 @@
 from mythril.laser.ethereum.state.annotation import StateAnnotation
 from mythril.laser.ethereum.svm import LaserEVM
 from mythril.laser.ethereum.plugins.signals import PluginSkipWorldState
+from mythril.laser.ethereum.plugins import LaserPlugin
 from mythril.laser.ethereum.state.global_state import GlobalState
 from mythril.laser.ethereum.transaction.transaction_models import (
     ContractCreationTransaction,
@@ -17,7 +18,7 @@ class MutationAnnotation(StateAnnotation):
     pass
 
 
-class MutationPruner:
+class MutationPruner(LaserPlugin):
     """Mutation pruner plugin
 
     Let S be a world state from which T is a symbolic transaction, and S' is the resulting world state.