Update tests for fix br0kage due to last commit