Fixed bug in migration 0002

Signed-off-by: Sheik Hassan <snh58@drexel.edu>