DeepSpeed

fixed a typo; this was fixed before but seems like it has been lost in the refactor

#364

Merged