Add missing db setup lines