fix typo and add mock