updated a few lines in test