Fixed assert errors

BTW, I'm no longer testing on Python 2.