slight fix of design