Revert "Remove top level nesting in test"

This reverts commit 79bb543ceec8855877b9dc3276622d8a08202669.