mini bug fix