Changed code so it could be tested. Just for a better code coverage...
Code coverage change