Late wickets restore equilibrium with Surrey