Update some files with small fix