Fix type. Closes #132