Fix test expected method