fixes problems with testing