Fix error message handling