fixed annoying bug

Program didn't work in Linux so I fixed the bug.