Updates config to use entity instead of model