Fix private bug in main module