Fixing issue during connection.