Add more migration stuff I forgot about