Change assertEquals to assertEqual