Fixed undefined variable

Fixes #14