fixing issue

Signed-off-by: Daniel Newns <dan@higgsdesign.com>