/* If MTL+ make sure gsc is loaded and proxy is setup */
        if (USE_HDCP_GSC(display)) {
-               if (!intel_hdcp_gsc_check_status(display))
+               if (!intel_hdcp_gsc_check_status(display->drm))
                        return false;
        }
 
 
        void *hdcp_cmd_out;
 };
 
-bool intel_hdcp_gsc_check_status(struct intel_display *display)
+bool intel_hdcp_gsc_check_status(struct drm_device *drm)
 {
-       struct drm_i915_private *i915 = to_i915(display->drm);
+       struct drm_i915_private *i915 = to_i915(drm);
        struct intel_gt *gt = i915->media_gt;
        struct intel_gsc_uc *gsc = gt ? >->uc.gsc : NULL;
 
        if (!gsc || !intel_uc_fw_is_running(&gsc->fw)) {
-               drm_dbg_kms(display->drm,
+               drm_dbg_kms(&i915->drm,
                            "GSC components required for HDCP2.2 are not ready\n");
                return false;
        }
        return err;
 }
 
-struct intel_hdcp_gsc_context *intel_hdcp_gsc_context_alloc(struct intel_display *display)
+struct intel_hdcp_gsc_context *intel_hdcp_gsc_context_alloc(struct drm_device *drm)
 {
-       struct drm_i915_private *i915 = to_i915(display->drm);
+       struct drm_i915_private *i915 = to_i915(drm);
        struct intel_hdcp_gsc_context *gsc_context;
        int ret;
 
         */
        ret = intel_hdcp_gsc_initialize_message(i915, gsc_context);
        if (ret) {
-               drm_err(display->drm, "Could not initialize gsc_context\n");
+               drm_err(&i915->drm, "Could not initialize gsc_context\n");
                kfree(gsc_context);
                gsc_context = ERR_PTR(ret);
        }
 
 #include <linux/err.h>
 #include <linux/types.h>
 
-struct intel_display;
+struct drm_device;
 struct intel_hdcp_gsc_context;
 
 ssize_t intel_hdcp_gsc_msg_send(struct intel_hdcp_gsc_context *gsc_context,
                                void *msg_in, size_t msg_in_len,
                                void *msg_out, size_t msg_out_len);
-bool intel_hdcp_gsc_check_status(struct intel_display *display);
+bool intel_hdcp_gsc_check_status(struct drm_device *drm);
 
-struct intel_hdcp_gsc_context *intel_hdcp_gsc_context_alloc(struct intel_display *display);
+struct intel_hdcp_gsc_context *intel_hdcp_gsc_context_alloc(struct drm_device *drm);
 void intel_hdcp_gsc_context_free(struct intel_hdcp_gsc_context *gsc_context);
 
 #endif /* __INTEL_HDCP_GCS_H__ */
 
 
        mutex_lock(&display->hdcp.hdcp_mutex);
 
-       gsc_context = intel_hdcp_gsc_context_alloc(display);
+       gsc_context = intel_hdcp_gsc_context_alloc(display->drm);
        if (IS_ERR(gsc_context)) {
                ret = PTR_ERR(gsc_context);
                kfree(arbiter);
 
 
 #define HDCP_GSC_HEADER_SIZE sizeof(struct intel_gsc_mtl_header)
 
-bool intel_hdcp_gsc_check_status(struct intel_display *display)
+bool intel_hdcp_gsc_check_status(struct drm_device *drm)
 {
-       struct xe_device *xe = to_xe_device(display->drm);
+       struct xe_device *xe = to_xe_device(drm);
        struct xe_tile *tile = xe_device_get_root_tile(xe);
        struct xe_gt *gt = tile->media_gt;
        struct xe_gsc *gsc = >->uc.gsc;
 }
 
 /*This function helps allocate memory for the command that we will send to gsc cs */
-static int intel_hdcp_gsc_initialize_message(struct intel_display *display,
+static int intel_hdcp_gsc_initialize_message(struct xe_device *xe,
                                             struct intel_hdcp_gsc_context *gsc_context)
 {
-       struct xe_device *xe = to_xe_device(display->drm);
        struct xe_bo *bo = NULL;
        u64 cmd_in, cmd_out;
        int ret = 0;
                                  XE_BO_FLAG_GGTT);
 
        if (IS_ERR(bo)) {
-               drm_err(display->drm, "Failed to allocate bo for HDCP streaming command!\n");
+               drm_err(&xe->drm, "Failed to allocate bo for HDCP streaming command!\n");
                ret = PTR_ERR(bo);
                goto out;
        }
        return ret;
 }
 
-struct intel_hdcp_gsc_context *intel_hdcp_gsc_context_alloc(struct intel_display *display)
+struct intel_hdcp_gsc_context *intel_hdcp_gsc_context_alloc(struct drm_device *drm)
 {
+       struct xe_device *xe = to_xe_device(drm);
        struct intel_hdcp_gsc_context *gsc_context;
        int ret;
 
         * NOTE: No need to lock the comp mutex here as it is already
         * going to be taken before this function called
         */
-       ret = intel_hdcp_gsc_initialize_message(display, gsc_context);
+       ret = intel_hdcp_gsc_initialize_message(xe, gsc_context);
        if (ret) {
-               drm_err(display->drm, "Could not initialize gsc_context\n");
+               drm_err(&xe->drm, "Could not initialize gsc_context\n");
                kfree(gsc_context);
                gsc_context = ERR_PTR(ret);
        }