1644 sålde familjen Königsberg fästningen till greve Adam Batthyány, som började en ombyggnad med syfte att omvandla fästningen till en representativ boning. Mellan 1647 och 1650 byggdes den västra flygeln och den tidigbarocka riddarsalen, borgens praktsal. Men även fästningens militära styrka visade sig bestå provet under det osmanska anfallet på 1680-talet och Rákócziupproret i början på 1700-talet.