Set price extra

Fix wrong usage of the BaseModel API and prevent KeyError with the context.