Missed an import for testing fix on Python 2.6