Fixed one more bug during compilation