trying to solve the bug