Fixing jQuery initialization