Corrected missing comma
Changing imports

- Other way didn't really make that much sense