Updated to work with nonreal answers