Fix a type error
Merge pull request #12 from alal/fix_type_error

Fix a type error