Fix jquery wrap