Fix dependancy issue