Handling issue: https://github.com/pycroscopy/atomai/issues/89

Handling issue: https://github.com/pycroscopy/atomai/issues/89 #313