Fixing changed function call