little fix on layout