Added missing changes needed for sucessfull migration.
Merge pull request #117 from davidschreiber/patch-1

Added missing changes needed for sucessfull migration.