Forgot to change test loop back to original loop