diff --git a/tensorflow_datasets/__init__.py b/tensorflow_datasets/__init__.py
index <HASH>..<HASH> 100644
--- a/tensorflow_datasets/__init__.py
+++ b/tensorflow_datasets/__init__.py
@@ -17,3 +17,7 @@
 # pylint: disable=g-multiple-import
 from tensorflow_datasets.core.dataset_builder import Split
 from tensorflow_datasets.core.registered import builder, registered, load
+
+# Imports for registration
+import tensorflow_datasets.image.cifar
+import tensorflow_datasets.image.mnist
diff --git a/tensorflow_datasets/image/cifar.py b/tensorflow_datasets/image/cifar.py
index <HASH>..<HASH> 100644
--- a/tensorflow_datasets/image/cifar.py
+++ b/tensorflow_datasets/image/cifar.py
@@ -20,12 +20,12 @@ from __future__ import division
 from __future__ import print_function
 
 import collections
-import cPickle
 import os
 import random
 
 import numpy as np
 import six
+from six.moves import cPickle
 import tensorflow as tf
 from tensorflow_datasets.core import dataset_builder
 from tensorflow_datasets.core import file_format_adapter
Import builder modules at top-level for registration

PiperOrigin-RevId: <I>