Fixes to datasets UI