Add additional upgrade check
Fixes #48