Fix build warning

Fixes issue #540