Fixing DNNHelper