Fix Controller

Signed-off-by: minhnh94 <minhnh.da@gmail.com>