Finally get coverage up to 100% :)