Fixed missing test headers