Turner, M. Clerc, V. Gorgisyan, I. Gschwendtner, E. Mazzoni, S. Petrenko, A.V. JACoW Geneva, Switzerland 978-3-95450-182-3 10.18429/JACoW-IPAC2017-TUPIK001 English 1682-1684 TUPIK001 proton plasma wakefield electron experiment Contribution to a conference proceedings 2017 2017-05 http://dx.doi.org/10.18429/JACoW-IPAC2017-TUPIK001 http://jacow.org/ipac2017/papers/tupik001.pdf The AWAKE project at CERN uses a self-modulated §I{400}{GeV/c} proton bunch to drive GV/m wakefields in a §I10{m} long plasma with an electron density of n_{pe} = 7 × 10¹⁴ \rm{electrons/cm}³. We present the upgrade of a proton beam diagnostic to indirectly prove that the bunch self-modulated by imaging defocused protons with two screens downstream the end of the plasma. The two-screen diagnostic has been installed, commissioned and tested in autumn 2016 and limitations were identified. We plan to install an upgraded diagnostics to overcome these limitations.