Update for latest DStack.
Temp commit.