Add missing import for py3_data decorator