fixing deprecation warning
fixing deprecation warning