Fix imports and syntax error