Fix matrix uses floats..... wtf numpy