use correct var
Merge pull request #176 from arpitgupta/working

use correct var