fixed test breaks