Fix test: fix typo in import statement