Fix in connection (add database variable)