Bug fix, missing variable declaration