chore(test): fix tests after module revert