Fix bug. Wrong arguments in function