Fix the type ref
Merge pull request #128 from raymondfeng/master

Fix the type ref