fix reference error issue #30

tested using setup as referenced in #30, verified all tests pass.