adding some tests back