Made some minor changes to middleware test.