Fixed the issue with scalability test