Fix accidentally undone change