Fixed array shape check
Removed unnecessary Kernel2D import