Whoops fixed logic