Fix location of wp_enqueue_media() call