refactor(test): code cleanup