Fix #311 - correct should syntax

Tiny fix for the warning on #311