Amendend spec as it was not returning the correct results