diff --git a/cmd/internal/markov/absorbing.go b/cmd/internal/markov/absorbing.go
index f975abe..306ca26 100644
--- a/cmd/internal/markov/absorbing.go
+++ b/cmd/internal/markov/absorbing.go
@@ -38,7 +38,7 @@ func NewAbsorbingMarkovChain() *AbsorbingMarkovChain {
 
 func (g *AbsorbingMarkovChain) IsValid() bool {
 	for nodes := g.Nodes(); nodes.Next(); {
-		u := nodes.Node().(*node)
+		u := nodes.Node()
 		if g.outWeightSum(u) != 1 {
 			return false
 		}
@@ -95,7 +95,7 @@ func (g *AbsorbingMarkovChain) AdjacencyMatrix() mat.Matrix {
 	nodes := adj.Nodes()
 	for i := 0; nodes.Next(); i++ {
 		id := nodes.Node().ID()
-		u := g.Node(id).(*node)
+		u := g.Node(id)
 
 		e := g.WeightedEdge(u.ID(), u.ID())
 		if e != nil {