diff --git a/aws_xray_sdk/core/context.py b/aws_xray_sdk/core/context.py index 553c85a2..0a2f395e 100644 --- a/aws_xray_sdk/core/context.py +++ b/aws_xray_sdk/core/context.py @@ -78,11 +78,13 @@ def end_subsegment(self, end_time=None): :param float end_time: epoch in seconds. If not specified the current system time will be used. """ - subsegment = self.get_trace_entity() - if self._is_subsegment(subsegment): - subsegment.close(end_time) + entity = self.get_trace_entity() + if self._is_subsegment(entity): + entity.close(end_time) self._local.entities.pop() return True + elif isinstance(entity, DummySegment): + return False else: log.warning("No subsegment to end.") return False