Update test so it works with py2.6