Fixed some compile errors in unit tests.