Minor fix to make it work on Firefox