Fix tests per #122