diff --git a/Android.bp b/Android.bp
index e9bf64e..d96e8dd 100644
--- a/Android.bp
+++ b/Android.bp
@@ -1,12 +1,17 @@
-override_android_library {
-    name: "pawlet-framework-overlay",
-    base: "framework",
-
-    remove_files: [
-        "core/java/android/os/UpdateEngine.java",
-    ],
-
-    add_files: [
+android_library {
+    name: "pawlet-framework-extension",
+    
+    srcs: [
         "frameworks/base/core/java/android/os/UpdateEngine.java",
     ],
+    
+    # Depend on the original framework
+    static_libs: [
+        "framework",
+    ],
+    
+    sdk_version: "current",
+    product_specific: true,
+    
+    overrides: ["framework"],
 }
\ No newline at end of file