FIXED: missing numpy import