Fix for bad import