Fix testing bug