minor tweak to DBN mnist dataset unit test