Updated unit test

Forgot that we already had one that could be easily grouped.