Add forgotten middleware