Use correct type

Fix compile warning