Remove unecesary atexit handler and tests for module.