Fixed formatting of online friends

1 files changed, 5 insertions, 3 deletions

diff --git a/Formatter.py b/Formatter.py
index 5cd065d..e43b782 100644
--- a/Formatter.py
+++ b/Formatter.py
@@ -58,10 +58,12 @@ def __formatExternalAuth(external_auth: fortnitepy.ExternalAuth):
     return '{} \({}\)'.format(external_auth.external_display_name, external_auth.type)
 
 def formatFriendOnline(display_name: str, party_size: int):
-    if party_size > 1:
-        text = 'is playing Fortnite with {} friends\!'.format(str(party_size - 1))
-    else:
+    if party_size == 1:
         text = 'is playing Fortnite\!'
+    elif party_size == 2:
+        text = 'is playing Fortnite with {} friend\!'.format(str(party_size - 1))
+    elif party_size > 2:
+        text = 'is playing Fortnite with {} friends\!'.format(str(party_size - 1))
     return formatting.format_text(
         u'\u2b50',
         formatting.mbold('{}'.format(display_name)),