Fixed DBN bug