small change in order to test mnist dataset