Fixed the bug caused by integration