Ok maybe this import was necessary after all