Fixed 2 argument issue