Fixing bug on the upgrade from 0.2 to 0.3