Fix jquery bug