Moar unit testing coverage