Tiny fix to the previous CL along the same lines
Merge pull request #507 from ussuri/fix-spark_widgets-imports-2

Tiny fix to the previous CL along the same lines