Last bit of unit testing