Update 0.6

Remove unnecessary import