minor fix fixes the bug