Fixed bad code from previous commit