fix typo, use newer code