Adding tests for upgrade method