Change to use *kocha.Controller

related commit: b0dc977ce1ed09d6ac225a98b8e59d46387abaf4