fix for shutdown and backpropagation