Fix error with imports