Fixing wrong variable reference.

Thanks for sharing this module, guys!