Fixes fault from previous commit