fix reference broken by refactoring