This shouldn’t be the case, as you’ve changed the number of input features.
Is the error message exactly the same and are you sure you’ve updated the code (make sure to rerun the cell in case you are using a Jupyter notebook).