Fix bug introduced in ce6f7c16d662a4618116bb321987dba2bdfe62a7. @phyllisstein