diff --git a/msm/dsi/dsi_display.c b/msm/dsi/dsi_display.c
index 91682051c5e2..38c478f94641 100644
--- a/msm/dsi/dsi_display.c
+++ b/msm/dsi/dsi_display.c
@@ -2716,7 +2716,7 @@ error:
 	return rc;
 }
 
-#ifdef CONFIG_DEEPSLEEP
+#if defined (CONFIG_DEEPSLEEP) || defined (CONFIG_HIBERNATION)
 int dsi_display_unset_clk_src(struct dsi_display *display)
 {
 	int rc = 0;