fix design a little bit