correcting bug in forward(size)