Fix error return
v.0.1.1