Added init method to migration callbacks