Fix import error
Merge remote-tracking branch 'upstream/master'