update code

Signed-off-by: qijiang.guo <qijiang.guo@gmail.com>
update for new controller.
Signed-off-by: qijiang.guo <qijiang.guo@gmail.com>