fixed future import ..., not yet fixed docstring