It doesn't look like `os` is used in this file. Can you remove this import?