World Model Robustness via Surprise Recognition