Added tests for updated comment migration