fixing tests for upgrade command with and without versions