DNN-5542 Fix tests