Fix testing code.