Fix bot: provide class on register