updated tests to not suck