Bug fix (input argument)