Fix tests broken after #98.