fix the bug of stage displacement