Fix error handling
0.0.6