Remove testing code from module