Bug fix, variable error