Fix some UI error on Layout Builder