Chage class name
Fix #1