Change according to Professor's comment
Signed-off-by: shunen.chen <shunen.chen@gmail.com>