diff --git a/greedy_method/greedy_knapsack.py b/greedy_method/greedy_knapsack.py
index 92dd81aaa..ed6399c74 100644
--- a/greedy_method/greedy_knapsack.py
+++ b/greedy_method/greedy_knapsack.py
@@ -15,10 +15,9 @@ weight[i] >= 0
 Calculate the maximum profit that the shopkeeper can make given maxmum weight that can
 be carried.
 """
-from typing import Union
 
 
-def calc_profit(profit: list, weight: list, max_weight: int) -> Union[str, int]:
+def calc_profit(profit: list, weight: list, max_weight: int) -> int:
     """
     Function description is as follows-
     :param profit: Take a list of profits