Fix for select all JS