Quick fix on update function (game context)