Fix datagrid test