Update to match v0.4 response changes