refactor(test): clean up tests