Fixed: test not adapted after refactoring