Adding missing middleware -__-