Quick fix for deprecation warning