diff --git a/deepchem/models/tests/test_mxmnet.py b/deepchem/models/tests/test_mxmnet.py
index 0a2d9d9bb4..b805ac270c 100644
--- a/deepchem/models/tests/test_mxmnet.py
+++ b/deepchem/models/tests/test_mxmnet.py
@@ -2,6 +2,7 @@
 import torch
 import tempfile
 import numpy as np
+import pytest
 from deepchem.feat.molecule_featurizers import MXMNetFeaturizer
 from deepchem.models.torch_models.mxmnet import MXMNet
 
@@ -11,6 +12,7 @@
 ]
 
 
+@pytest.mark.torch
 def test_mxmnet_regression():
     """
     Test MXMNet class for regression