Fix account tests