Fix bug in watch mode.