Removed unneeded import that was breaking on travis