Fix minor design bug