Fix Deprecation warning