Fix KeyError on embedding missing field