diff --git a/inference.py b/inference.py
index c32d7cf..a1c721b 100644
--- a/inference.py
+++ b/inference.py
@@ -231,6 +231,7 @@ def main():
 	while 1:
 		start_idx = int(i * mel_idx_multiplier)
 		if start_idx + mel_step_size > len(mel[0]):
+			mel_chunks.append(mel[:, len(mel[0]) - mel_step_size:])
 			break
 		mel_chunks.append(mel[:, start_idx : start_idx + mel_step_size])
 		i += 1