Fix using wheel