[New Feature] - Fix postgres connection issue