Fix migration for comments