Fix the unit tests
Merge pull request #6 from wilsonge/layouts-field

Fix the unit tests :)