Backwards compatibility of load_data (~ problem).
Accidentally moved a line after it was needed.