Looks like I need to fix the update method too.