Small layout tweak