* fix problem on upgrade