diff --git a/aptos-move/aptos-e2e-comparison-testing/src/lib.rs b/aptos-move/aptos-e2e-comparison-testing/src/lib.rs
index ee2b3e21506ea..02e0d01620828 100644
--- a/aptos-move/aptos-e2e-comparison-testing/src/lib.rs
+++ b/aptos-move/aptos-e2e-comparison-testing/src/lib.rs
@@ -32,7 +32,7 @@ pub use data_collection::*;
 pub use execution::*;
 use move_compiler::compiled_unit::CompiledUnitEnum;
 use move_core_types::language_storage::ModuleId;
-use move_model::metadata::CompilerVersion;
+use move_model::metadata::{CompilerVersion, LanguageVersion};
 use move_package::{
     compilation::compiled_package::CompiledPackage,
     source_package::{
@@ -449,6 +449,7 @@ fn compile_package(
     let mut build_options = aptos_framework::BuildOptions {
         compiler_version,
         bytecode_version: Some(7),
+        language_version: Some(CompilerVersion::infer_stable_language_version(&compiler_version.unwrap_or_default())),
         ..Default::default()
     };
     build_options
diff --git a/aptos-move/aptos-e2e-comparison-testing/src/main.rs b/aptos-move/aptos-e2e-comparison-testing/src/main.rs
index 4a3c6d75230fc..bd4f20e47a86c 100644
--- a/aptos-move/aptos-e2e-comparison-testing/src/main.rs
+++ b/aptos-move/aptos-e2e-comparison-testing/src/main.rs
@@ -43,6 +43,9 @@ pub enum Cmd {
          /// Sampling rate (1-10)
         #[clap(long, default_value_t = SAMPLING_RATE)]
         rate: u32,
+        /// Branch of framework
+        #[clap(long)]
+        branch: Option<String>,
     },
     /// Collect and execute txns without dumping the state data
     Online {
@@ -124,7 +127,8 @@ async fn main() -> Result<()> {
             skip_source_code_check: skip_source_code,
             dump_write_set,
             target_account,
-            rate
+            rate,
+            branch
         } => {
             let batch_size = BATCH_SIZE;
             let output = if let Some(path) = output_path {
@@ -136,7 +140,7 @@ async fn main() -> Result<()> {
                 std::fs::create_dir_all(output.as_path()).unwrap();
             }
             if !skip_source_code {
-                prepare_aptos_packages(output.join(APTOS_COMMONS), None).await;
+                prepare_aptos_packages(output.join(APTOS_COMMONS), branch).await;
             }
             let data_collector = DataCollection::new_with_rest_client(
                 Client::new(Url::parse(&endpoint)?),