fixed bug where only working on iterations of first model