Small fix to resolve warnings