Changing part of code that give build error