Fixing up compiler warning