class fix design
class fix design