Fix pep error

I'll self merge...