Tiny fix on layout