Fixes UI docs