diff --git a/seqlearn/hmm.py b/seqlearn/hmm.py
index 59df882..b7dfacb 100644
--- a/seqlearn/hmm.py
+++ b/seqlearn/hmm.py
@@ -64,7 +64,7 @@ def fit(self, X, y, lengths):
 
         init_prob = np.log(Y[start].sum(axis=0) + alpha)
         init_prob -= logsumexp(init_prob)
-        final_prob = np.log(Y[start].sum(axis=0) + alpha)
+        final_prob = np.log(Y[end - 1].sum(axis=0) + alpha)
         final_prob -= logsumexp(final_prob)
 
         feature_prob = np.log(safe_sparse_dot(Y.T, X) + alpha)