Fix an issue due to import aliasing

One cell fails to work because numpy was imported as "np"