Fix a type and close issue #2