rewrite control metthod

Signed-off-by: Grarak <williye97@gmail.com>