fix training layout