fix design a bit