fixed the bug :)