fix setting of momentum