Jessica Chastain wanted Johan Heldenbergh to play the lead role in The Zookeeper's Wife