Fixed rabbit unit test after refactoring