Import missing FlatFit from pypiezo.surface.

diff --git a/pyafm/afm.py b/pyafm/afm.py
index b03ad05b2c0b4f99f75dedd6acd1891eafdb4f47..bae7e3947eb2536395c7352ec4d4a68b17a37a59 100644 (file)
--- a/pyafm/afm.py
+++ b/pyafm/afm.py
@@ -26,6 +26,7 @@ module only contains methods that require the capabilities of both.
 from pypiezo.base import convert_bits_to_meters as _convert_bits_to_meters
 from pypiezo.base import convert_meters_to_bits as _convert_meters_to_bits
 from pypiezo.base import convert_volts_to_bits as _convert_volts_to_bits
+from pypiezo.surface import FlatFit as _FlatFit
 
 from . import LOG as _LOG