Test fix, middleware changed names