Check against expected lines of output

Stop checking for the hard coded value "restart", instead pass in the
expected amount of output.