extend main class tests