fixed small last minute bug