change import path for wrap and wrap-contrib in tests