Fix testing issue