diff --git a/psychrnn/tasks/perceptual_discrimination.py b/psychrnn/tasks/perceptual_discrimination.py
index 1652851..c52ba7e 100644
--- a/psychrnn/tasks/perceptual_discrimination.py
+++ b/psychrnn/tasks/perceptual_discrimination.py
@@ -112,7 +112,7 @@ def trial_function(self, t, params):
             x_t[dir] += 1 + coh
             x_t[(dir + 1) % 2] += 1
 
-        if t > onset + stim_dur + 20:
+        if t >= onset + stim_dur:
             y_t[dir] = self.hi
             y_t[1-dir] = self.lo