Small fix db connection