Fix additional logic that hadn't been transitioned to new name
Merge pull request #43 from theponti/develop

Fix additional logic that hadn't been transitioned to new name