Bug fix, incorrect variable name
0.3.1