Fix label
Add error handling