Fix compile for unit tests