Temporairly hack into main model