Use forked version of onMediaQuery that includes callback args