diff --git a/auto_ml/predictor.py b/auto_ml/predictor.py
index 111f02e..cb4937c 100644
--- a/auto_ml/predictor.py
+++ b/auto_ml/predictor.py
@@ -1003,7 +1003,7 @@ def print_results(self, model_name, model, X, y):
 
                 feature_responses = self.create_feature_responses(model, X, y, top_features)
                 self._join_and_print_analytics_results(feature_responses, sorted_model_results, sort_field='Importance')
-            except TypeError as e:
+            except AttributeError as e:
                 if model_name == 'XGBRegressor':
                     pass
                 else: