Make module definition compatible with Python 2 as well.