add back part of test