Corrected the bug on firefox