Fix for previous change

Forgot to remove part of the previous code