diff --git a/tests/sweep_framework/sweeps/matmul/full/matmul_default_height_sharded.py b/tests/sweep_framework/sweeps/matmul/full/matmul_default_height_sharded.py
index 44cd92fe128c..fff716bc2ea1 100644
--- a/tests/sweep_framework/sweeps/matmul/full/matmul_default_height_sharded.py
+++ b/tests/sweep_framework/sweeps/matmul/full/matmul_default_height_sharded.py
@@ -88,7 +88,7 @@ def run(
     # TODO: row_wise=False and ROW_MAJOR shard orientation gives bad PCC
     # TODO: COL_MAJOR shard orientation doesn't work for get_matmul_program_config
     input_a_memory_config.shard_spec = ttnn.ShardSpec(
-        ttnn.num_cores_to_corerange_set(num_cores_height, core_grid, row_wise=True),
+        ttnn.CoreRangeSet(ttnn.num_cores_to_corerange_set(num_cores_height, core_grid, row_wise=True)),
         (per_core_height, k_size),
         ttnn.ShardOrientation.ROW_MAJOR,
         False,
diff --git a/tests/sweep_framework/sweeps/matmul/full/matmul_default_width_sharded.py b/tests/sweep_framework/sweeps/matmul/full/matmul_default_width_sharded.py
index e6fb7562639f..a7f0361f4fe5 100644
--- a/tests/sweep_framework/sweeps/matmul/full/matmul_default_width_sharded.py
+++ b/tests/sweep_framework/sweeps/matmul/full/matmul_default_width_sharded.py
@@ -81,7 +81,7 @@ def run(
     # TODO: row_wise=False and ROW_MAJOR shard orientation gives bad PCC
     # TODO: COL_MAJOR shard orientation doesn't work for get_matmul_program_config
     input_a_memory_config.shard_spec = ttnn.ShardSpec(
-        ttnn.num_cores_to_corerange_set(num_cores_width, core_grid, row_wise=True),
+        ttnn.CoreRangeSet(ttnn.num_cores_to_corerange_set(num_cores_width, core_grid, row_wise=True)),
         (total_height, per_core_width),
         ttnn.ShardOrientation.ROW_MAJOR,
         False,
diff --git a/tests/sweep_framework/sweeps/matmul/short/matmul_default.py b/tests/sweep_framework/sweeps/matmul/short/matmul_default.py
index d872fcf1874d..bb12db78c588 100644
--- a/tests/sweep_framework/sweeps/matmul/short/matmul_default.py
+++ b/tests/sweep_framework/sweeps/matmul/short/matmul_default.py
@@ -12,7 +12,7 @@
 from models.utility_functions import torch_random
 
 
-TIMEOUT = 5
+TIMEOUT = 15
 
 # TODO: Missing coverage for mixed precision; passed in dtype does nothing in current matmul path
 parameters = {