Add benchmark test back