Fixing arguments error found in tests