Changed Model. Should be correctly working now.