Added tests for future migrations