Fixed reference-error from recent restructuring