Minor fix to make it runable