Small bug fix (import statement)