Another hotfix this time adding a seperate update method.