Restore accidentally removed test