After the Second World War, the former State Railway lines became part of the Österreichische Bundesbahnen (ÖBB). The rolling stock was given a new number scheme in 1953. In the following years, there were some alignment corrections on the line. That, and the rebuilding of the rolling stock in the 1950s and 1960s, as well as the changeover to diesel working on the branch line were the furthest-reaching modernization measures undertaken on the railway. Between 1959 and 1962, the original electric locomotives, now known as ÖBB class 1099, were given new, more modern style, bodies, whilst retaining the same traction equipment. The passenger coaches were likewise provided with matching steel bodies. In 1962, the steam engines were replaced by new diesel locomotives, of ÖBB class 2095. The series 399 locomotives went to the Waldviertler Schmalspurbahnen, whilst the remaining steam locomotives were withdrawn from service.