Fixing build warning