Minor changes..

Couple of quick fixes in the testing script and the module..