Fix minor issue in account migration