fix type error
Merge pull request #35 from srethink/fix_type_error

fix type error