Forgot to commit some changes.

Signed-off-by: Xiaomao Chen <xiaomao5@live.com>