diff --git a/django_fsm/management/commands/graph_transitions.py b/django_fsm/management/commands/graph_transitions.py index e21c98a..9ce6da7 100644 --- a/django_fsm/management/commands/graph_transitions.py +++ b/django_fsm/management/commands/graph_transitions.py @@ -105,9 +105,9 @@ def generate_dot(fields_data): final_states = targets - sources for name, label in final_states: - subgraph.node(name, label=label, shape="doublecircle") + subgraph.node(name, label=str(label), shape="doublecircle") for name, label in (sources | targets) - final_states: - subgraph.node(name, label=label, shape="circle") + subgraph.node(name, label=str(label), shape="circle") if field.default: # Adding initial state notation if label == field.default: initial_name = node_name(field, "_initial")