diff --git a/CIFAR10_playground.py b/CIFAR10_playground.py
index ef8c6f1..c725731 100755
--- a/CIFAR10_playground.py
+++ b/CIFAR10_playground.py
@@ -20,10 +20,10 @@ train_dset = torchvision.datasets.CIFAR10(root='./CIFAR10',train=True,download=F
 test_dset = torchvision.datasets.CIFAR10(root='./CIFAR10',train=False,download=False,transform=transforms.ToTensor())
 train_loader = torch.utils.data.DataLoader(train_dset, batch_size=128, shuffle=True, num_workers=0)
 test_loader = torch.utils.data.DataLoader(test_dset, batch_size=128, shuffle=False, num_workers=0)
-train_dset.train_data.to(device)
-train_dset.train_labels.to(device)
-test_dset.test_data.to(device)
-test_dset.test_labels.to(device)
+train_dset.data.to(device)
+train_dset.target.to(device)
+test_dset.data.to(device)
+test_dset.target.to(device)
 #######################################################