{% for p in players.keys() %}
diff --git a/templates/cwa/set.html.j2 b/templates/cwa/set.html.j2
index 56aa300..56c8ac4 100644
--- a/templates/cwa/set.html.j2
+++ b/templates/cwa/set.html.j2
@@ -52,7 +52,12 @@
{%- endfor %}
- Refresh
+
+ {% if game.is_completed %}
+ Return to lobby
+ {% else %}
+ Refresh
+ {% endif %}
{%- endblock %}
diff --git a/webgames/go.py b/webgames/go.py
index ccda7b5..9d877f9 100644
--- a/webgames/go.py
+++ b/webgames/go.py
@@ -281,7 +281,7 @@ Press Play! to play against GNU Go or invite a second player.
@property
def is_completed(self) -> bool:
- return self.black_passed and self.white_passed
+ return (self.black_passed and self.white_passed) or self._abandoned
GoPuzzle()