Fixed connection to DB