just a litte design change fix