Fix stupid logic