diff --git a/common-npm-packages/docker-common/package-lock.json b/common-npm-packages/docker-common/package-lock.json
index 1e08995e..06791695 100644
--- a/common-npm-packages/docker-common/package-lock.json
+++ b/common-npm-packages/docker-common/package-lock.json
@@ -1,12 +1,12 @@
 {
     "name": "azure-pipelines-tasks-docker-common",
-    "version": "2.264.0",
+    "version": "2.265.0",
     "lockfileVersion": 2,
     "requires": true,
     "packages": {
         "": {
             "name": "azure-pipelines-tasks-docker-common",
-            "version": "2.264.0",
+            "version": "2.265.0",
             "license": "MIT",
             "dependencies": {
                 "@types/mocha": "^5.2.7",
diff --git a/common-npm-packages/docker-common/package.json b/common-npm-packages/docker-common/package.json
index 4766e58b..c1e30ddb 100644
--- a/common-npm-packages/docker-common/package.json
+++ b/common-npm-packages/docker-common/package.json
@@ -1,6 +1,6 @@
 {
     "name": "azure-pipelines-tasks-docker-common",
-    "version": "2.264.0",
+    "version": "2.265.0",
     "description": "Common Library for Azure Rest Calls",
     "repository": {
         "type": "git",
diff --git a/common-npm-packages/docker-common/pipelineutils.ts b/common-npm-packages/docker-common/pipelineutils.ts
index e4ac844b..6b45d774 100644
--- a/common-npm-packages/docker-common/pipelineutils.ts
+++ b/common-npm-packages/docker-common/pipelineutils.ts
@@ -55,45 +55,19 @@ function addReleaseLabels(hostName: string, labels: string[], addPipelineData?:
 
 function addBaseImageLabels(connection: ContainerConnection, labels: string[], dockerFilePath: string): void {
 
-    //Added Dynamic feature flag to fix getting digest image from docker file
-    let digestImageFromFileEnabled = tl.getPipelineFeature('UseDigestImageFromFile');
-
-    if (digestImageFromFileEnabled) {
-
-        // using getBaseImageDetialsFromDockerFIle method to fetch both image and imagedigest
-        const baseImage = containerUtils.getBaseImageDetialsFromDockerFIle(dockerFilePath, connection);
-        if (!baseImage.name) {
-            return;
-        }
-        addLabelWithValue("image.base.ref.name", baseImage.name, labels);
-        //first check if there is digest passed in ]Dockerfile
-        if (!baseImage.digest) {
-            baseImage.digest = containerUtils.getImageDigest(connection, baseImage.name);
-        }
-        //if there is no digest in Dockerfile, get digest using ImageName:tag
-        if (baseImage.digest) {
-            addLabelWithValue("image.base.digest", baseImage.digest, labels);
-        }
-
+    // using getBaseImageDetialsFromDockerFIle method to fetch both image and imagedigest
+    const baseImage = containerUtils.getBaseImageDetialsFromDockerFIle(dockerFilePath, connection);
+    if (!baseImage.name) {
+        return;
     }
-    else {
-        const baseImageName = containerUtils.getBaseImageNameFromDockerFile(dockerFilePath);
-        if (!baseImageName) {
-            return;
-        }
-        addLabelWithValue("image.base.ref.name", baseImageName, labels);
-
-        if (!connection) {
-            tl.debug("Image digest couldn't be extracted because no connection was found.");
-            return;
-        }
-
-        const baseImageDigest = containerUtils.getImageDigest(connection, baseImageName);
-
-        //if there is no digest in Dockerfile, get digest using ImageName:tag
-        if (baseImageDigest) {
-            addLabelWithValue("image.base.digest", baseImageDigest, labels);
-        }
+    addLabelWithValue("image.base.ref.name", baseImage.name, labels);
+    //first check if there is digest passed in Dockerfile
+    if (!baseImage.digest) {
+        baseImage.digest = containerUtils.getImageDigest(connection, baseImage.name);
+    }
+    //if there is no digest in Dockerfile, get digest using ImageName:tag
+    if (baseImage.digest) {
+        addLabelWithValue("image.base.digest", baseImage.digest, labels);
     }
 }