Small bug, need to change a variable