Quick change to the code to see if it works