Fixed type that causes issue in latest 0.2.2 release