Remove reference to data which contains wpdb + updated test