Fix preprocess_data script.