Arguments passed to probe function are correct now