Moved test under main module