Fix one line bug