Added a preprocessor directive where required.