diff --git a/graphics/composer/aidl/android/hardware/graphics/composer3/vts/functional/composer-vts/ReadbackVts.cpp b/graphics/composer/aidl/android/hardware/graphics/composer3/vts/functional/composer-vts/ReadbackVts.cpp
index e832cb7fc5..b612309138 100644
--- a/graphics/composer/aidl/android/hardware/graphics/composer3/vts/functional/composer-vts/ReadbackVts.cpp
+++ b/graphics/composer/aidl/android/hardware/graphics/composer3/vts/functional/composer-vts/ReadbackVts.cpp
@@ -274,7 +274,7 @@ LayerSettings TestColorLayer::toRenderEngineLayerSettings() {
 
 TestBufferLayer::TestBufferLayer(const std::shared_ptr<IComposerClient>& client,
                                  const ::android::sp<::android::GraphicBuffer>& graphicBuffer,
-                                 TestRenderEngine& renderEngine, int32_t display, uint32_t width,
+                                 TestRenderEngine& renderEngine, int64_t display, uint32_t width,
                                  uint32_t height, common::PixelFormat format,
                                  Composition composition)
     : TestLayer{client, display}, mRenderEngine(renderEngine) {
diff --git a/graphics/composer/aidl/android/hardware/graphics/composer3/vts/functional/composer-vts/include/ReadbackVts.h b/graphics/composer/aidl/android/hardware/graphics/composer3/vts/functional/composer-vts/include/ReadbackVts.h
index 58989d8286..85b4fdce2d 100644
--- a/graphics/composer/aidl/android/hardware/graphics/composer3/vts/functional/composer-vts/include/ReadbackVts.h
+++ b/graphics/composer/aidl/android/hardware/graphics/composer3/vts/functional/composer-vts/include/ReadbackVts.h
@@ -53,7 +53,7 @@ class TestRenderEngine;
 
 class TestLayer {
   public:
-    TestLayer(const std::shared_ptr<IComposerClient>& client, int32_t display)
+    TestLayer(const std::shared_ptr<IComposerClient>& client, int64_t display)
         : mComposerClient(client) {
         client->createLayer(display, kBufferSlotCount, &mLayer);
     }
@@ -100,7 +100,7 @@ class TestLayer {
 
 class TestColorLayer : public TestLayer {
   public:
-    TestColorLayer(const std::shared_ptr<IComposerClient>& client, int32_t display)
+    TestColorLayer(const std::shared_ptr<IComposerClient>& client, int64_t display)
         : TestLayer{client, display} {}
 
     void write(const std::shared_ptr<CommandWriterBase>& writer) override;
@@ -117,7 +117,7 @@ class TestBufferLayer : public TestLayer {
   public:
     TestBufferLayer(const std::shared_ptr<IComposerClient>& client,
                     const ::android::sp<::android::GraphicBuffer>& graphicBuffer,
-                    TestRenderEngine& renderEngine, int32_t display, uint32_t width,
+                    TestRenderEngine& renderEngine, int64_t display, uint32_t width,
                     uint32_t height, common::PixelFormat format,
                     Composition composition = Composition::DEVICE);