Fix bug in imports