Refactor test case
Redo migration file
Add test for module