Fix 0.6 compatibility

This appears to fix #49.