correct tests such that all code is tested