diff --git a/ccrestoration/model/vsr_base_model.py b/ccrestoration/model/vsr_base_model.py
index f714dc6..3c11a23 100644
--- a/ccrestoration/model/vsr_base_model.py
+++ b/ccrestoration/model/vsr_base_model.py
@@ -51,7 +51,7 @@ def inference_image_list(self, img_list: List[np.ndarray]) -> List[np.ndarray]:
 
             for i in range(out.shape[1]):
                 img = out[0, i, :, :, :]
-                img = img.squeeze(0).permute(1, 2, 0).cpu().numpy()
+                img = img.permute(1, 2, 0).cpu().numpy()
                 img = (img * 255).clip(0, 255).astype("uint8")
                 img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
                 res_img_list.append(img)
diff --git a/ccrestoration/vs/vsr.py b/ccrestoration/vs/vsr.py
index 1082250..9056d69 100644
--- a/ccrestoration/vs/vsr.py
+++ b/ccrestoration/vs/vsr.py
@@ -59,7 +59,7 @@ def _inference(n: int, f: list[vs.VideoFrame]) -> vs.VideoFrame:
             output = inference(img)
 
             for i in range(output.shape[0]):
-                cache[n + i] = output[i, :, :, :]
+                cache[n + i] = output[0, i, :, :, :]
 
         return _tensor_to_frame(cache[n], f[1].copy())
 
diff --git a/pyproject.toml b/pyproject.toml
index 3ab2899..f50272a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -40,7 +40,7 @@ license = "MIT"
 name = "ccrestoration"
 readme = "README.md"
 repository = "https://github.com/TensoRaws/ccrestoration"
-version = "0.0.6"
+version = "0.0.7"
 
 # Requirements
 [tool.poetry.dependencies]