Small fix on design