fixing class migration bug
Merge pull request #30 from maraoz/bug/p2p-quickfix

fixing class migration bug