From dba094475c9231f4b66f4b17f5aadced58cecec2 Mon Sep 17 00:00:00 2001
From: Sander Roosendaal <roosendaalsander@gmail.com>
Date: Wed, 28 Nov 2018 11:46:40 +0100
Subject: [PATCH] small improvement

---
 rowers/plannedsessions.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/rowers/plannedsessions.py b/rowers/plannedsessions.py
index 6cf62bda..04f6c194 100644
--- a/rowers/plannedsessions.py
+++ b/rowers/plannedsessions.py
@@ -825,7 +825,12 @@ def race_can_withdraw(r,race):
     return True
 
 def race_can_register(r,race):
-    records = VirtualRaceResult.objects.filter(
+    if race.sessiontype == 'race':
+        recordobj = VirtualRaceResult
+    else:
+        recordobj = IndoorVirtualRaceResult
+        
+    records = recordobj.objects.filter(
         userid=r.id,
         race=race)