Fix NADE unittests