Fix broken if