Fix deprecation warning in base wrapper