Unit test compile fix