fix: forgotten directive after refactoring