Added a bit to the Unit Tests