From ecb26e9eec76ff7415b4fcc7b072a58f8469c87c Mon Sep 17 00:00:00 2001
From: Thomas Wiecki <thomas.wiecki@gmail.com>
Date: Thu, 6 Dec 2012 16:07:18 -0500
Subject: [PATCH] BUG: Can not test for length when dropping nans.

---
 zipline/test_algorithms.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/zipline/test_algorithms.py b/zipline/test_algorithms.py
index 21c4b62c..99f92a02 100644
--- a/zipline/test_algorithms.py
+++ b/zipline/test_algorithms.py
@@ -299,7 +299,7 @@ class BatchTransformAlgorithm(TradingAlgorithm):
             fillna=False
         )
 
-        self.return_nan = ReturnPriceBatchTransform(
+        self.return_nan = return_price_batch_decorator(
             refresh_period=self.refresh_period,
             window_length=self.window_length,
             fillna=True