Fix after_create deprecation warning