Adding missing header on test class