Le , incapable d'empêcher la cavalerie polonaise de rallier Stefan Czarniecki et sous la pression de ses subalternes, Jan Fryderyk abandonne le camp suédois et combat désormais dans l'armée de Czarniecki. Il participe ainsi à la bataille de Varsovie et aux actions contre le prince de Transylvanie Georges II Rákóczi (1657), le blocus de la Prusse (1658-1659).