Fix build error

Credit: Bajee