NotMnist fixes