Fix bug in import