Fix deprecation warning

Closes #11
Closes #10