fix loading cleaned dataset