fix tests after new design