Little layout fix