fix error in layout()
Merge pull request #157 from d1rtym0nk3y/develop

fix error in layout()