Fix compilation issue

Fix compilation issue