Fixed mini bug