Modified test acoording to prevoius commit