diff --git a/dice_ml/explainer_interfaces/dice_genetic.py b/dice_ml/explainer_interfaces/dice_genetic.py
index 0b1bc1b7..b491eb1b 100644
--- a/dice_ml/explainer_interfaces/dice_genetic.py
+++ b/dice_ml/explainer_interfaces/dice_genetic.py
@@ -76,7 +76,7 @@ def do_loss_initializations(self, yloss_type, diversity_loss_type, feature_weigh
                         feature_weights_list.append(feature_weights[feature])
                     else:
                         # the weight is inversely proportional to max value
-                        feature_weights_list.append(round(1 / self.feature_range[feature].max(), 2))
+                        feature_weights_list.append(round(1 / max(self.feature_range[feature]), 2))
             self.feature_weights_list = [feature_weights_list]
 
     def do_random_init(self, num_inits, features_to_vary, query_instance, desired_class, desired_range):