Add test for head check