Fixed wrong type
Fixes #11