0.5.1 fixes