Fix missing layout