Fix build. Update connection string for stage