correct fix for the bug