diff --git a/examples/2d/cifar_small_sample.py b/examples/2d/cifar_small_sample.py
index e63422ed7..02ad62e46 100644
--- a/examples/2d/cifar_small_sample.py
+++ b/examples/2d/cifar_small_sample.py
@@ -9,7 +9,7 @@
 import torch.optim
 from torchvision import datasets, transforms
 import torch.nn.functional as F
-from kymatio.torch import Scattering2D
+from kymatio import Scattering2D
 import torch
 import argparse
 import kymatio.datasets as scattering_datasets