Fix imports to get the performance tests compiling.