Fix import of codegen templates

Actually tis should be:

```
from .templates import
```