use correct parameter from main class