Small dependency fix