bug fix

forgot to import numpy