Test the original implementation works