Adding middleware back into the story