Unit tests could be a lot more comprihensive and cover a lot more of the code. When someone has time this would be a great improvement