Fixing v0.1.7 bugs