Remove after_initialize block