Fixing test with new header