added unit test(s) for the driver module and resolved some issues that came along while running the test
minor tweak