Fixes

Fixes #198
Fixes #201