add tests back in