Minor change to fix compilation issues