Fix testing logic.