quick fix for shape errors