Sgd fixes