Fix bug when return if pattern