chore(tests): minor cleanup in tests