Fix AttributeError: 'super' object has no attribute 'capture_traffic'