Updating tests to include a mock header