minor fix design