From 972662e4ae8d0ef66c54ec090fc2b73bfc260406 Mon Sep 17 00:00:00 2001
From: Nikhil Parasaram <tommycjniko@gmail.com>
Date: Mon, 25 Feb 2019 15:35:38 +0530
Subject: [PATCH] Fix typehints in MythrilAnalyzer

---
 mythril/interfaces/cli.py           | 2 --
 mythril/mythril/mythril_analyzer.py | 4 ++--
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/mythril/interfaces/cli.py b/mythril/interfaces/cli.py
index 6bad208a..e448a34b 100644
--- a/mythril/interfaces/cli.py
+++ b/mythril/interfaces/cli.py
@@ -63,7 +63,6 @@ def create_parser(parser: argparse.ArgumentParser) -> None:
     """
     Creates the parser by setting all the possible arguments
     :param parser: The parser
-    :return:
     """
     parser.add_argument("solidity_file", nargs="*")
 
@@ -256,7 +255,6 @@ def parse_args(parser: argparse.ArgumentParser, args: argparse.Namespace) -> Non
     Parses the arguments
     :param parser: The parser
     :param args: The args
-    :return:
     """
     if args.epic:
         path = os.path.dirname(os.path.realpath(__file__))
diff --git a/mythril/mythril/mythril_analyzer.py b/mythril/mythril/mythril_analyzer.py
index 669ebbdb..6441eb3a 100644
--- a/mythril/mythril/mythril_analyzer.py
+++ b/mythril/mythril/mythril_analyzer.py
@@ -45,7 +45,7 @@ class MythrilAnalyzer:
     def dump_statespace(
         self,
         strategy: str,
-        contract: List[EVMContract] = None,
+        contract: EVMContract = None,
         address: Optional[str] = None,
         max_depth: Optional[int] = None,
         execution_timeout: Optional[int] = None,
@@ -84,7 +84,7 @@ class MythrilAnalyzer:
         self,
         strategy: str,
         address: str,
-        contract: List[EVMContract] = None,
+        contract: EVMContract = None,
         max_depth: Optional[int] = None,
         enable_physics: bool = False,
         phrackify: bool = False,