diff --git a/shared/source/os_interface/linux/xe/ioctl_helper_xe.h b/shared/source/os_interface/linux/xe/ioctl_helper_xe.h
index b2c6c334fb..3129ab1e49 100644
--- a/shared/source/os_interface/linux/xe/ioctl_helper_xe.h
+++ b/shared/source/os_interface/linux/xe/ioctl_helper_xe.h
@@ -157,7 +157,7 @@ class IoctlHelperXe : public IoctlHelper {
     };
 
     uint16_t getDefaultEngineClass(const aub_stream::EngineType &defaultEngineType);
-    void setContextProperties(const OsContextLinux &osContext, void *extProperties, uint32_t &extIndexInOut);
+    virtual void setContextProperties(const OsContextLinux &osContext, void *extProperties, uint32_t &extIndexInOut);
 
     int maxExecQueuePriority = 0;
     std::mutex xeLock;
diff --git a/shared/source/os_interface/os_context.h b/shared/source/os_interface/os_context.h
index d6155c99c7..97c9eadba9 100644
--- a/shared/source/os_interface/os_context.h
+++ b/shared/source/os_interface/os_context.h
@@ -82,7 +82,7 @@ class OsContext : public ReferenceTrackedObject<OsContext> {
     void setContextGroup(bool value) {
         isContextGroup = value;
     }
-    bool isPartOfContextGroup() {
+    bool isPartOfContextGroup() const {
         return isContextGroup;
     }