Improve error message handling