Fixes model spec