Update after migrations