Fixed build error in previous commit