diff --git a/config/model/pluto_model.yaml b/config/model/pluto_model.yaml
index 49145d5..fdf09a5 100644
--- a/config/model/pluto_model.yaml
+++ b/config/model/pluto_model.yaml
@@ -16,7 +16,9 @@ num_modes: 12
 state_dropout: 0.75
 use_ego_history: false
 state_attn_encoder: true
-use_hidden_proj: false
+use_hidden_proj: true
+cat_x: true
+ref_free_traj: true
 
 feature_builder:
   _target_: src.feature_builders.pluto_feature_builder.PlutoFeatureBuilder