Fix Layout Builder JS issue