Fixes compile error after refactor