Amend tests based on feedback