last minute bug fix!

last minute bug fix!