Fixes test logic, was missing exit calls.