Put back missing lines of code.