Fixed assertRaises syntax for Python 2.6