Fix wrong class name ;)