diff --git a/src/test/java/com/yello/server/domain/purchase/FakePurchaseRepository.java b/src/test/java/com/yello/server/domain/purchase/FakePurchaseRepository.java
index dbd7b948..06e71d2a 100644
--- a/src/test/java/com/yello/server/domain/purchase/FakePurchaseRepository.java
+++ b/src/test/java/com/yello/server/domain/purchase/FakePurchaseRepository.java
@@ -4,6 +4,7 @@
 
 import com.yello.server.domain.purchase.entity.ProductType;
 import com.yello.server.domain.purchase.entity.Purchase;
+import com.yello.server.domain.purchase.entity.PurchaseState;
 import com.yello.server.domain.purchase.exception.PurchaseNotFoundException;
 import com.yello.server.domain.purchase.repository.PurchaseRepository;
 import com.yello.server.domain.user.entity.User;
@@ -92,8 +93,8 @@ public Purchase getEndByStateAndUserId(User user) {
         return data.stream()
             .filter(purchase -> {
                 return purchase.getUser().equals(user) &&
-                    purchase.getProductType().equals("yello_plus") &&
-                    purchase.getState().equals("active");
+                    purchase.getProductType().equals(ProductType.YELLO_PLUS) &&
+                    purchase.getState().equals(PurchaseState.ACTIVE);
             })
             .sorted(Comparator.comparing(Purchase::getUpdatedAt).reversed())
             .findFirst()