Update to get it working using Elli's init approach to middleware. (yay)