Fix rescue clause