Fixing '. was unexpected at this time error'