Fixes the bug