Fixed compile bug in previous commit