From a4632783fba26aaaedd0b9777877cb2ec8b48752 Mon Sep 17 00:00:00 2001
From: Erik Johnston <erik@matrix.org>
Date: Tue, 8 Nov 2016 11:20:26 +0000
Subject: [PATCH] Sample correctly

---
 synapse/handlers/message.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/synapse/handlers/message.py b/synapse/handlers/message.py
index 3eca46df86..81df45177a 100644
--- a/synapse/handlers/message.py
+++ b/synapse/handlers/message.py
@@ -425,7 +425,9 @@ class MessageHandler(BaseHandler):
 
                 # We also randomly point to some of the older events, to make
                 # sure that we don't completely ignore the older events.
-                new_latest_ret.extend(random.sample(latest_ret, 5))
+                if latest_ret[5:]:
+                    sample_size = min(5, len(latest_ret[5:]))
+                    new_latest_ret.extend(random.sample(latest_ret[5:], sample_size))
                 latest_ret = new_latest_ret
 
             if latest_ret: