Fixed bug (shape of result matrix)