diff --git a/tests/modules/layers/test_transformer.py b/tests/models/flava/test_transformer.py
similarity index 99%
rename from tests/modules/layers/test_transformer.py
rename to tests/models/flava/test_transformer.py
index 5c5d1ebf..a4c3ffa1 100644
--- a/tests/modules/layers/test_transformer.py
+++ b/tests/models/flava/test_transformer.py
@@ -9,7 +9,7 @@
 import torch
 from tests.test_utils import assert_expected, set_rng_seed
 from torch import nn
-from torchmultimodal.modules.layers.transformer import (
+from torchmultimodal.models.flava.transformer import (
     TransformerEncoder,
     TransformerEncoderLayer,
 )