I had the same thought. Not so much that he'd be treated worse in the US, but that the Chinese police seemed perfectly reasonable given the newness/strangeness of the "offense".

I was also pleasantly surprised he got the copter and camera back at the end.