Update tests to not break.