putting tests back in