Unit action menu bug fix