Fix compability with clean_data.