Add couple of error handling tests