Fixed a mini-bug

Update by Zalthrion