Added db context. Hopefully this will work correctly.