Fix for lecture 7 example.