SGD Update fixes