Fix migration warnings