fixing dataset mnist