small improvment about layout