Fixes a wrong import statement.

Signed-off-by: Markus Hubig <mhubig@gmail.com>