Small bug fix

Should stop some client errors.