Hot fix

Small patch with little fix that I forgot to do on previous update ;)