diff --git a/cacheflow/decoding.py b/cacheflow/decoding.py
index c4c4a13b..143f0ab2 100644
--- a/cacheflow/decoding.py
+++ b/cacheflow/decoding.py
@@ -1,7 +1,7 @@
 from typing import Optional, Set
 
 
-class DecodingParams:
+class SamplingParams:
 
     def __init__(
         self,