bug fixing

again the same bug. I'm stupid.