Readd a couple of missing lines