Fixed the soundness bug!