Small compilation error from refactoring fixed