small fix
last commit from wenhao