Added compatibility for 0.2 (untested)

As suggested by @timholy