Cleaned up migration stub