fix some missing code.