Minor change with UI