fixed testing issue