fixes problem related to main class