Fix missing variable introduced during refactoring