fix a small warning

Still have a small warning