Modify the following Python code to raise an exception if the input value is not an integer
def square_root(x):
    return x**0.5
def square_root(x):
    if not isinstance(x, int):
        raise TypeError('Input must be an integer')
    return x**0.5