small last minute bug fix