some mini bug fixes