Identify the bug in this code
Input: def add_two_numbers(a, b):
    total = a + b + 1
    return total
The issue with this code is that the function is adding the number 1 to both the 'a' and 'b' variables, which is not what is desired. Instead, the code should only add 1 to the variable 'total', which is the sum of the 'a' and 'b' variables.
The bug in this code is that the value of 'total' is not stored in the variable 'a'. The correct way to write this code is: 'def add_two_numbers(a, b): a = a + b + 1; return a'. This way, the variable 'a' will store the updated value of 'total'.