Offset Keyframes 3.0.1 for Maya (maya script)

Hey Charlie, I was wondering if there was a quick fix for getting this script to work in Maya 2017? It works so good in 16 and below and I would love to continue using it?

thanks

Jon