fix var assignment
Merge pull request #3 from arkxu/master

fix var assignment