Toto Wolff has admitted his surprise at Mercedes' major upgrade package for the Bahrain Grand Prix having appeared to take the team backwards, following an underwhelming qualifying display.