Fix testing regression