fix for DNN issue