Fixed compiler error from last commit