Fixes Migration Unit Test