smaller bug fix