Modification

Signed-off-by: Wu Xudong <tikiet@gmail.com>