Fix build for tests