fixed the bug ....