This should fix compatibility

Let me know if it doesn't work