Quick last minute bug fix