Fix call to assertRaises