Fix: Minor change