Fix datatypes

Fixes #5