small error handling change