Fixing load_data task