Removed unnecesary piece of code in test