Unit dialog design issue