Restore normalize_plain_observation function