corrected tests to run correctly