Fix code.
Merge pull request #1 from lambdafu/master

Fix code.