diff --git a/tupak/result.py b/tupak/result.py index afd7cfda30a1fabf860d3d686a2eb1568ca99def..d5da239e248b40c91b85fe88bc4d3e7ebd85c195 100644 --- a/tupak/result.py +++ b/tupak/result.py @@ -154,7 +154,7 @@ class Result(dict): kwargs['truths'] = kwargs.pop('truth') if getattr(self, 'injection_parameters', None) is not None: - injection_parameters = [self.injection_parameters[key] + injection_parameters = [self.injection_parameters.get(key, None) for key in self.search_parameter_keys] kwargs['truths'] = kwargs.get('truths', injection_parameters)