Quick fix on UI

Changed some attributes to reflect changes to data type