Add the lines in order to fix bug with db