Remove 'update gains' hack