Re-instate essential line removed in refactoring