Fixing build warning/error