Update package to be 0.1.1, fix broken tests