adding 2 line in test