Fix based on @peterflynn's feedback.