Re-add test