#157 Fixing migration