Fixed assertRaises errors