slight ui fix