Fixes to collocation unit tests

Fixes to collocation unit tests.