Instead of hardcoding e.g. flatten output, we should use the snapbox test framework