Add missing var.
Merge pull request #2 from evanlucas/hotfix/missing-var

Add missing var.