fix for 0.9.1

Hey

I just updated it for 0.9.1