Small edition of ui to work with new arch, fixing.