Added fix for pypy support.