Added lacking header in the test unit