fix error with simple example.