update tests for new design