Fix broken tests
Add headers argument to get