fix testing issue