From 58c4f7840a9767fae917aa5374a29f428653a237 Mon Sep 17 00:00:00 2001
From: Brian Wo <45139213+brainwo@users.noreply.github.com>
Date: Thu, 29 Aug 2024 17:40:40 +0800
Subject: [PATCH] Add missing difficulty adjusment

---
 lib/src/fsrs_base.dart | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/lib/src/fsrs_base.dart b/lib/src/fsrs_base.dart
index e2ffbb8..a32912c 100644
--- a/lib/src/fsrs_base.dart
+++ b/lib/src/fsrs_base.dart
@@ -159,6 +159,11 @@ class FSRS {
     double retrievability,
     State state,
   ) {
+    s.again.difficulty = _nextDifficulty(lastD, Rating.again);
+    s.hard.difficulty = _nextDifficulty(lastD, Rating.hard);
+    s.good.difficulty = _nextDifficulty(lastD, Rating.good);
+    s.easy.difficulty = _nextDifficulty(lastD, Rating.easy);
+
     switch (state) {
       case State.learning:
       case State.relearning: