Fix bug when find_layout called with string