fix AttributeError: 'NoneType' object has no attribute 'data'